我下面有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})
答案 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()