Files
Exiled-Exchange-2/dataParser/vendor/client/services/runes.py
T

38 lines
1.1 KiB
Python

import pandas as pd
def get_df(json_file: dict) -> pd.DataFrame:
return pd.DataFrame(json_file)
def build_runes_df(
base_items: pd.DataFrame, stats: pd.DataFrame, runes: pd.DataFrame
) -> pd.DataFrame:
def replace_indices_with_ids(index_list, stats_df):
return [stats_df.loc[i, "Id"] for i in index_list]
runes["StatsArmour"] = runes["StatsArmour"].apply(
lambda lst: replace_indices_with_ids(lst, stats)
)
runes["StatsMartialWeapon"] = runes["StatsMartialWeapon"].apply(
lambda lst: replace_indices_with_ids(lst, stats)
)
runes["StatsCasterWeapon"] = runes["StatsCasterWeapon"].apply(
lambda lst: replace_indices_with_ids(lst, stats)
)
runes_merged = runes.merge(
base_items, left_on="BaseItemType", right_index=True, how="left"
)[
[
"BaseItemType",
"StatsMartialWeapon",
"StatsValuesMartialWeapon",
"StatsArmour",
"StatsValuesArmour",
"StatsCasterWeapon",
"StatsValuesCasterWeapon",
]
]
return runes_merged