Numpy. select并在df中指定新列,条件为两列的值

x7rlezfr  于 2022-12-18  发布在  其他
关注(0)|答案(1)|浏览(101)
df  = df.assign[test  = np.select[df.trs = 'iw' & df.rp == 'yu'],[1,0],'null']

我想要if df.trs == iw'和df.rp == ' yu ''那么新列应该被创建应该是0否则1仅仅是为了填充行而不是每一行
我尝试了no.slect和条件数组。但没有得到所需的输出

sy5wg1nm

sy5wg1nm1#

您不需要numpy.select,一个简单的布尔运算符就足够了:

df['test'] = (df['trs'].eq('iw') & df['rp'].eq('yu')).astype(int)

如果你真的想使用numpy,这将需要numpy.where

df['test'] = np.where(df['trs'].eq('iw') & df['rp'].eq('yu'), 1, 0)

相关问题