python - 将多列 Pandas 数据框与一列进行比较

Question

我有一个数据框:
df-

    A   B   C   D   E
0   V   10  5   18  20
1   W   9   18  11  13
2   X   8   7   12  5
3   Y   7   9   7   8
4   Z   6   5   3   90

我想添加一列“结果”，如果“E”列中的值大于 B、C 和 D 列中的值，则应返回 1，否则返回 0。

输出应该是:

    A   B   C   D   E   Result
0   V   10  5   18  20  1
1   W   9   18  11  13  0
2   X   8   7   12  5   0
3   Y   7   9   7   8   0
4   Z   6   5   3   90  1

对于几列，我会使用类似的逻辑:if(and(E>B,E>C,E>D),1,0),
但是我必须将大约 20 列(从 B 到 U)与名称“V”进行比较。此外，数据框有大约 10 万行。

我在用

df['Result']=np.where((df.ix[:,1:20])<df['V']).all(1),1,0)

它给出了内存错误。

Answer 1

一种可能的解决方案是在 numpy 中进行比较最后将 bool 掩码转换为 int s:

df['Result'] = (df.iloc[:, 1:4].values < df[['E']].values).all(axis=1).astype(int)
print (df)
   A   B   C   D   E  Result
0  V  10   5  18  20       1
1  W   9  18  11  13       0
2  X   8   7  12   5       0
3  Y   7   9   7   8       0
4  Z   6   5   3  90       1