找到pandas中某些列的总和

Question

我正在尝试使用pandas来对某些列进行求和，同时保留其他列。 例如：

member_no, data_1, data_2, data_3, dat_1, dat_2, other_1, other_2

1, 1, 3, 0, 0, 1, 1, 0

1, 1, 3, 0, 0, 1, 0, 1

2, 0, 1, 5, 1 ,0, 1, 0

2, 0, 1, 5, 1 ,0, 0, 1

我想要结果

member_no, data_1, data_2, data_3, dat_1, dat_2, other_1, other_2  

1, 1, 3, 0, 0, 1, 1, 1

2, 0, 1, 5, 1, 0, 1, 1

对于给定的成员id，具有'data'和'dat'的所有列将具有相同的值，因此我只想保留它。 具有“其他”属性的列需要求和。

谢谢您的帮助。

Answer 1

你在member_no + max上寻找一个groupby 。

df = df.groupby('member_no', as_index=False).max()
print(df)
   member_no  data_1  data_2  data_3  dat_1  dat_2  other_1  other_2
0          1       1       3       0      0      1        1        1
1          2       0       1       5      1      0        1        1