如何在Python中的熊猫中添加缺少值的列

Question

我有一个熊猫数据框，如下所示。

   name    x1     x2
0  a       5      5
1  b       8      6
2  e       3      8

我还有一个names的列表，如下所示。

lst = [a, e]

我想在当前数据框中添加一个名为final_values的列；

• 如果name在lst中，则列值应为1
• 其他0

因此，我更新的数据框应如下所示。

   name    x1     x2    final_values
0  a       5      5        1
1  b       8      6        0
2  e       3      8        1

在熊猫中有什么简单的方法吗？

很高兴在需要时提供更多详细信息。

Answer 1

通过Series.isin检查成员身份，并将True/False到1/0映射将布尔值转换为整数：

df['final_values'] = df['name'].isin(lst).astype(int)

或使用numpy.where：

df['final_values'] = np.where(df['name'].isin(lst), 1, 0)