python - 如何在 python 中的 pandas 中添加具有缺失值的列

Question

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

   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

在 pandas 中有什么简单的方法可以做到这一点吗？

如果需要，我很乐意提供更多详细信息。

Answer 1

通过 Series.isin 查看成员(member)资格并将 True/False 的 bool 值转换为整数到 1/0 映射:

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

或者使用numpy.where :

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