def prepare_data(self, data: DataProvider): df_all = data.get_all_transactions().copy() df_all.loc[df_all[config.CATEGORY_SUB_COL] == '', config.CATEGORY_SUB_COL] = config.MISC_CATEGORY for main_category in config.categories.keys(): df_cat = df_all.loc[ df_all[config.CATEGORY_MAIN_COL] == main_category, [config.CATEGORY_SUB_COL, config.ABSAMOUNT_COL]] self._x_axis, sub_category_values = data.aggregate_by_category_as_tuple( df_cat, 'MS', config.CATEGORY_SUB_COL, main_category) self._main_category_values[main_category] = sub_category_values
def prepare_data(self, data: DataProvider): df_all = data.get_all_transactions() self._x_axis, self._category_values = data.aggregate_by_category_as_tuple( df_all, 'YS', config.CATEGORY_MAIN_COL)