python - Pandas:具有多个条件的 numpy.where 逻辑

Question

Python 数据科学新手

下面是我的Raw_data

raw_data = {'var1': ['true','false','ture'],'var2': [10,20,50], 'var3':['eggs','milk','eggs']}
df = pd.DataFrame(raw_data, columns = ['var1','var2','var3'])`

已测试代码但不起作用

def my_fun (var1,var2,var3,var4):
    df[var4]= np.where((df[var1] == 'true', 
                       df[var3] == 'eggs',
                       df[var2] < 10),
                       'hello',
                       'zello')
return df

这里我喜欢使用var1、var2和var3条件并获得条件结果。请帮忙

Answer 1

首先使用 bool 值 True/False 而不是字符串来简化逻辑。将此转换应用于系列'var1':

df['var1'] = df['var1'] == 'true'

然后您可以使用按位运算符&来比较 bool 系列:

def my_fun (var1,var2,var3,var4):
    df[var4]= np.where(df[var1] & df[var3].eq('eggs') & df[var2].lt(10),
                       'hello', 'zello')
    return df

效率较低的替代方案是使用np.ological_and.reduce:

def my_fun (var1,var2,var3,var4):
    conds = (df[var1], df[var3] == 'eggs', df[var2] < 10)
    df[var4]= np.where(np.logical_and.reduce(conds), 'hello', 'zello')
    return df