如何仅对具有给定列表中值的行添加列并设置值

Question

假设我有一个像这样的数据框：

   ID   size    price
0   1   5      300
1   2   10     500
2   3   20     600
3   4   35     800
4   5   65     900
5   6   70     1000

并且我想向其中添加一列“已删除”，对于给定列表中的“ ID”，它将有1个，否则为0。

ids_wanted = [1,4,6]

结果将是：

    ID    size    price   removed
0   1      5       300        1
1   2     10       500        0
2   3     20       600        0
3   4     35       800        1
4   5     65       900        0
5   6     70      1000        1

我的代码是：

import numpy as np
import pandas as pd
i = range(1,7)
s = [5,10,20,35,65,70]
p = [300,500,600,800,900,1000]
df = pd.DataFrame({"ID":i,"size":s,"price":p})
ids_wanted = [1,4,6]
df['removed']=['1','0','0','1','0','1']
df

Answer 1

如果您想用0或1动态标记它们，则需要将对'removed'列的声明更改为类似的内容：