python - 根据列值和列表创建列

Question

我有一个 df，我想在其中创建一个新列，其值基于列表和另一列之间的匹配。

这是模拟 df:

df = pd.DataFrame(
{'fruit': ['Apple', 'Banana', 'Clementine', 'Dragon fruit', 'Elderberry', 'Figs', 'Grapes'],
'bites': [1, 2, 3, 1, 2, 3, 4]})

有了这个列表:

good = ['Apple', 'Clementine', 'Figs', 'Grapes']

我尝试创建所需的列:

df['good']= np.where(df['fruit']== good, 'good', 'not good')

这就是我想要实现的。请注意，我更喜欢此文本而不是 bool 值。

Answer 1

使用Series.isin检查成员(member)资格:

df['good']= np.where(df['fruit'].isin(good), 'good', 'not good')
print (df)
          fruit  bites      good
0         Apple      1      good
1        Banana      2  not good
2    Clementine      3      good
3  Dragon fruit      1  not good
4    Elderberry      2  not good
5          Figs      3      good
6        Grapes      4      good