python - 如何求和引用其他数据框的总数

Question

我想对两个 DataFrame 的所有组合求和，

DataFrame A

         ColA    ColB   Sales
           1       A      10
           1       B      20
           1       C      100
           2       D      1000
           2       E      2000

DataFrame B
         ColA    ColB   
          1      A,B
          2      E

我对 DataFrame 的预期结果是

        ColA     ColB   TotalSales
         1       A,B     30
         2       E       2000

Answer 1

您可以尝试选择位于数据框 B ColB 列中的数据框 A 行，并添加所选的销售列

#df
 ColA    ColB   Sales
   1       A      10
   1       B      20
   1       C      100
   2       D      1000
   2       E      2000


df.set_index('ColB',inplace=True)
#df
#       ColA    Sales
# ColB      
# A        1    10
# B        1    20
# C        1    100
# D        2    1000
# E        2    2000

#df1

 #ColA    ColB   
 # 1      A,B
 # 2      E

df1['TotalSales'] = df1.ColB.str.split(',').apply(lambda x: df.loc[x]['Sales'].sum() )

输出:

ColA    ColB    TotalSales
0   1   A,B 30
1   2   E   2000