Example #1
0
    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)