保留数据框中与列表相符的行。

Question

我有一个独特元素列表和一个包含多列的数据帧。我希望保留数据帧中也存在于该列表中的所有行。

例如，考虑以下列表和数据帧/ CSV文件：

列表：[a, c, e]

数据帧：

ColA    ColB
  a       1
  b       2
  c       3
  d       4
  e       5
  f       6

我希望修改数据框如下：

ColA    ColB
  a       1
  c       3
  e       5

我该如何做呢？我希望使用 ColA 中唯一的值作为依据来保留或删除元素。

- Tavish Jain

Answer 1

使用 Series.isin 方法：

In [2597]: df
Out[2597]: 
  ColA  ColB
0    a     1
1    b     2
2    c     3
3    d     4
4    e     5
5    f     6

In [2599]: l = ['a', 'c', 'e']

In [2602]: df = df[df['ColA'].isin(l)]

In [2603]: df
Out[2603]: 
  ColA  ColB
0    a     1
2    c     3
4    e     5

- Mayank Porwal