python - 获取列具有条件的 Pandas 数据框的指定索引

Question

我有一个 Pandas 数据框:

id    value
14     122
15     120
16     190
17     490
18     328
19     309
20     323

我有一个 id 的列表 L = [14,17,20] 并且想做两件事:

Get the list of indexes of those rows where for the id's is not in the list L

i.e.( index of rows 15,16,18,19)

和，

Delete the rows for the id's is not in the list L.

我的预期输出:

id    value
14     122
17     490
20     323

Answer 1

您可以使用 boolean indexing与 isin , 用于反转 bool 掩码 ~:

idx = df.index[~df['id'].isin(L)].tolist()
print (idx)
[1, 2, 4, 5]

然后 drop :

df1 = df.drop(idx)
print (df1)
   id  value
0  14    122
3  17    490
6  20    323

备选方案:

df1 = df[df['id'].isin(L)]
print (df1)
   id  value
0  14    122
3  17    490
6  20    323