pandas df 表中 想把某列中相同元素 的其他列中元素求和
import pandas as pd
df = pd.DataFrame(columns= ["code","sum","+-"])
for i in range(5):
df.loc[i]=["001",i+1,"+"]
for i in range(5,10):
df.loc[i]=["001",i+1,"-"]
for i in range(10,15):
df.loc[i]=["002",i+1,"+"]
print(df)
code sum +-
0 001 1 +
1 001 2 +
2 001 3 +
3 001 4 +
4 001 5 +
5 001 6 -
6 001 7 -
7 001 8 -
8 001 9 -
9 001 10 -
10 002 11 +
11 002 12 +
12 002 13 +
13 002 14 +
14 002 15 +
想得到
code sum
001 -25 # 1+2+3+4+5-6-7-8-9
002 65 # 11+12+13+14+15
应该怎么做呢
002
是不是可以把它放进一个字典里面,就是先创建一个空字典,遍历的时候判断键名是否在字典中,不存在则以code的值为键名,初始值为0加第一个sum(带上正负符号),存在则进行累加。