python - 在数据框中添加依赖于现有列的列的通用公式

Question

我有两个数据框，如下所示。

数据框1 =

   ID   D   F   G   Q
0   T1  1   3   4   5
1   T2  2   3   2   6
2   T3  4   7   7   8
3   T4  5   6   7   8

数据框2 =

    ID  Q   G   F   
0   T10 10  54  44  
1   T12 20  45  43  
2   T11 30  45  65  

我想通过以下方法添加一列。

dataframe1['sum'] = dataframe1.D + dataframe1.F + dataframe1.G + dataframe1.Q

这将为 dataframe1 提供 sum 列，但如果我对 dataframe2 应用相同的公式，则会抛出错误，因为dataframe2 中没有列 D。

我想要一个通用公式，即使 D 不存在，也能给出 sum(在这种情况下，只需忽略 D 并考虑剩下的那些)根据公式计算列并计算总和)。

我怎样才能用一行代码做到这一点？

Answer 1

df.sum 将添加所有数字列并忽略具有 object 数据类型的列

df["sum"] = df.sum(axis=1)