如何将列表索引映射到数据框索引并标记匹配的位置

Question

我有以下列表

  [4,5,6]

我有以下数据框

  index    quantity
  0        12
  1        45
  2        56
  3        55
  4        65
  5        43
  6        23
  7        34

我想在数据框中与数据框索引匹配的新列中标记列表索引。

 index    quantity    flag
  0        12         0
  1        45         0
  2        56         0
  3        55         0
  4        65         1
  5        43         1
  6        23         1 
  7        34         0

如何在大熊猫中做到这一点？

Answer 1

将Index.isin用作布尔掩码并将其转换为整数-True/False s到1/0 s：

L = [4,5,6]
df['flag'] = df.index.isin(L).astype(int)
print (df)
       quantity  flag
index                
0            12     0
1            45     0
2            56     0
3            55     0
4            65     1
5            43     1
6            23     1
7            34     0