大PandasGroupby和Aggreg

2024-06-12 10:21:56 发布

您现在位置:Python中文网/ 问答频道 /正文

下面的groupby/agg组合做了两件奇怪的事情:

  1. 它从原始数据帧复制列
  2. 它返回所有列,而不仅仅是我在agg函数中特别强调的那些列。你知道吗

在应用函数之前,我尝试过重置索引和排序。那没什么帮助。你知道吗

df_final = df_initial.groupby('ID').agg({'MASTER_MEMBER_ID':'first' ,
                                         'pdc':'mean' , 
                                         'patient_pay':'sum' , 
                                         'DRUG_NAME':'first' , 
                                         'DAYS_SUPPLY':'sum' , 
                                         'PAID_AMT':'sum', 
                                         'total_days':'mean' , 
                                         'covered_days':'mean', 
                                         'end_date':'first', 
                                         'LIS_CAT':'first', 
                                         'YEAR':'first'})

我只希望看到我在agg函数中调用的列的列表,当然不希望看到重复的列。你知道吗


Tags: 函数iddf原始数据排序mean事情days