как сопоставить индексы списка с индексом фрейма данных и пометить, где он соответствует

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