python - 根据数据框中的相似列在数据框中添加列

Question

我试图获得一个输出，我希望在 d1 和 d2 中添加列 d ，其中 a b c 相同(如 groupby)。

例如

d1 = pd.DataFrame([[1,2,3,4]],columns=['a','b','c','d'])

d2 = pd.DataFrame([[1,2,3,4],[2,3,4,5]],columns=['a','b','c','d'])

然后我想得到一个输出

   a  b  c  d
0  1  2  3  8
1  2  3  4  5

合并两个数据框并添加结果列d，其中a b c 相同。

d1.add(d2) 或 radd 为我提供所有列的聚合

解决方案应该是一个可以类似地再次添加到另一个 DataFrame 的 DataFrame。

感谢任何帮助。

Answer 1

您可以使用set_index第一:

print (d2.set_index(['a','b','c'])
         .add(d1.set_index(['a','b','c']), fill_value=0)
         .astype(int)
         .reset_index())

   a  b  c  d
0  1  2  3  8
1  2  3  4  5