Groupby大熊猫抛出ValueError:石斑鱼和轴的长度必须相同

时间:2020-04-28 04:51:44

标签: python pandas pandas-groupby

我下面有csv

ID,PR_No,PMO,PRO,REV,COST
111,111,AB,MA,2575,2575
111,111,AB,MA,-1137,-1137
112,112,CD,KB,1134,3334
111,111,AB,MA,100,100

输出如下

ID,PR_No,PMO,PRO,REV,COST
111,111,AB,MA,1538,1538
112,112,CD,KB,1134,3334

1538 = 2575-1137 + 100

我的代码抛出值错误

df_n = df.groupby([['ID','PR_No','PMO','PRO']]).agg({'REV':sum,'COST':sum})

2 个答案:

答案 0 :(得分:2)

删除列名列表的嵌套[]

df_n = df.groupby(['ID','PR_No','PMO','PRO']).agg({'REV':sum,'COST':sum})
print (df_n)
                    REV  COST
ID  PR_No PMO PRO            
111 111   AB  MA   1538  1538
112 112   CD  KB   1134  3334

因为可以使用.sum在groupby之后创建相同的聚合函数列表:

df_n = df.groupby(['ID','PR_No','PMO','PRO'])['REV','COST'].sum()

ValueError:石斑鱼和轴的长度必须相同

是什么意思?

如果使用示例数据运行,则可以正常工作,因为嵌套列表的长度与示例数据的长度相同:

df_n = df.groupby([['ID','PR_No','PMO','PRO']]).agg({'REV':sum,'COST':sum})
print (df_n)
        REV  COST
ID     2575  2575
PMO    1134  3334
PRO     100   100
PR_No -1137 -1137

如果创建具有重复项的列表,它将聚合示例数据:

df_n = df.groupby([['ID','ID','PRO','PRO']]).agg({'REV':sum,'COST':sum})
print (df_n)
      REV  COST
ID   1438  1438
PRO  1234  3434

但是如果长度不同,则会失败:

print (df)
    ID  PR_No PMO PRO   REV  COST
0  111    111  AB  MA  2575  2575
1  111    111  AB  MA -1137 -1137
2  112    112  CD  KB  1134  3334
3  111    111  AB  MA   100   100
4  111    111  AB  MA   100   100 <- added new row

df_n = df.groupby([['ID','ID','PRO','PRO']]).agg({'REV':sum,'COST':sum})
print (df_n)

ValueError:石斑鱼和轴的长度必须相同

答案 1 :(得分:-1)

尝试使用一些agg或sum函数,它将起作用...

df_n = df.groupby([......])[....].sum()