Pandas уменьшение размера исходного фрейма данных при создании сводной таблицы с помощью функции? - PullRequest
0 голосов
/ 06 мая 2020

Я создал pandas фрейм данных, который включает несколько методов groupby. Длина df в порядке. Однако когда я отправляю фрейм данных в созданную мной функцию, он резко сокращает его размер, и я не совсем понимаю, почему. Пожалуйста, посмотрите мой код ниже.

def return_pivot(df, cols_for_use, index_col, agg_type, incl_margins=False, dupe_drop_cols=[]):

   dfc = df

   if dupe_drop_cols:
       dfc.drop_duplicates(subset=dupe_drop_cols, inplace=True)

   dfc = dfc[cols_for_use]

   return dfc.pivot_table(index=index_col, aggfunc=agg_type, margins=incl_margins).fillna(0).astype('int')

pv1 = return_pivot(df=posture, index_col=['field1', 'field2'], agg_type='sum', incl_margins=True, cols_for_use=state_mkt_ppl)

pv2 = return_pivot(df=posture, index_col='field1', agg_type=pd.Series.nunique, incl_margins=True, cols_for_use=state_fn, dupe_drop_cols=['field5'])
...