有没有办法将 np.where() 与列列表一起使用？

Question

我尝试使用 np.where() 函数创建新变量。myDF['newVar'] = np.where((myDF['var1']==1) |                           (myDF['var2']==1) |                           (myDF['var3']==1) ,                           1, 0)有没有办法用这样的列表替换 var1、var2、var3 （每列相同的条件 ==1）：listVars=['var1', 'var2', 'var3']myDF['newVar'] = np.where((myDF[listVars]==1),                           1, 0)

Answer 1

你可以用pandas.DataFrame.any()与axis=1：

listVars=['var1', 'var2', 'var3']

myDF['newVar'] = np.where((myDF[listVars]==1).any(axis=1), 1, 0)

例如：

myDF = pd.DataFrame({

    "var1": [1, 1, 1, 0, 0],

    "var2": [1, 0, 1, 0, 1],

    "var3": [0, 0, 0, 0, 0]

})

listVars=['var1', 'var2', 'var3']

myDF['newVar'] = np.where((myDF[listVars]==1).any(1),1, 0)

print(myDF)

#   var1  var2  var3  newVar

#0     1     1     0       1

#1     1     0     0       1

#2     1     1     0       1

#3     0     0     0       0

#4     0     1     0       1