ifelse从熊猫的相应专栏获取价值

Question

问题背后的想法是得到这样的东西：

data['output'] = np.where(data['column']=='Z', 0, 1)

但是在我的情况下，我真正需要的不是特定的0或1值，而是另一个列中的值，例如：

data['output'] = np.where(data['column']=='Z', data['col1'], data['col2'])

不确定是否可以这样

Answer 1

您可以尝试以下代码：

TRUE