在数据框中查找值并在相应列中交叉引用值

Question

我有一个数据框如下：

Index   X_1  X_2  X_3  W_1   W_2   W_3
1       IEZ  XOP  ABC  0.42  0.18  0.40
2       PXJ  ABC  XES  0.47  0.12  0.41
3       ABC  RYE  PXE  0.23  0.33  0.44
4       XOP  IEZ  ABC  0.62  0.20  0.18

我想检查X\u1到X\u3列之间每个“ABC”实例的索引行。对于每一个实例，我都想从W\u1到W\u3列中提取相应的值，并将其附加到一个新的W列中

例如，对于索引行1，如果ABC的一个实例出现在xu3中，它将从wu3中提取相应的值。你知道吗

最终产品应该是这样的。你知道吗

Index   X_1  X_2  X_3  W_1   W_2    W_3     **W**
1       IEZ  XOP  ABC  0.42  0.18   0.40    **0.40**
2       PXJ  ABC  XES  0.47  0.12   0.41    **0.12**
3       ABC  RYE  PXE  0.23  0.33   0.44    **0.23**
4       XOP  IEZ  ABC  0.62  0.20   0.18    **0.18**

如何仅在导入numpy和pandas模块的情况下执行此操作？

Answer 1

另一种方法：

df = pd.DataFrame({'X_1' : ['IEZ', 'PXJ', 'ABC', 'XOP'],  
                   'X_2' : ['XOP', 'ABC', 'RYE', 'IEZ'], 
                   'X_3' : ['ABC', 'XES','PXE', 'ABC'],
                   'W_1' :  [0.42, 0.47, 0.23, 0.62],
                   'W_2' : [0.18, 0.12, 0.33, 0.20],
                   'W_3' :  [0.40, 0.41, 0.44, 0.18]})

首先，取数字列：

num_columns = df.loc[:,'W_1':'W_3']

接下来，使用X_1->；X_3列生成布尔掩码：

df_mask = (df.loc[:,'X_1':'X_3']=='ABC').values

最后，使用DataFrame mask方法，当单元为真时返回NaNs，当掩码为假时返回单元值。然后，我们将对结果行求和，并将其分配给原始数据帧：

df['W'] = num_columns.mask(~df_mask).sum(axis=1)

当然，这可以组合成一行：

df['W'] = (df.loc[:,'W_1':'W_3']
            .mask(~(df.loc[:,'X_1':'X_3']=='ABC').values)
            .sum(axis=1))

编辑：

当然，这只在每行只有一个'ABC'实例的情况下才有效-您可能需要对此进行检查。你知道吗