python - Pandas:如何将多个模式(OR)与 np.where 匹配

Question

我想知道pandas中的np.where是否可行使用“OR”参数匹配多个模式

例如，我尝试在 DataFrame 中创建一个名为“kind”的新列并为每一行填充“测试”，如果另一列中的值称为“标签”匹配任何列出的模式，否则用“control”填充。

我正在使用这个:

df['kind'] = np.where(df['label'] == 'B85_C', 'test', 'control')

它在 1 种模式下运行良好

我正在寻找的是这样的:

df['kind'] = np.where(df['label'] == 'B85_C'OR'B85_N' ,'test', 'control')

有什么想法如何执行该操作或者是否有替代方案？谢谢

Answer 1

您可以使用按位或:

(df['label'] == 'B85_C') | (df['label'] == 'B85_N')

或者您可以使用 isin方法:

df['label'].isin(['B85_C', 'B85_N'])