如何从dataframe中选出存在列表中的元素呢

假设有以下的数据框 df 和列表 lst：

import pandas as pd

df = pd.DataFrame({'A': ['apple', 'banana', 'carrot'], 'B': [10, 20, 30]})
lst = ['apple', 'cat', 'banana']

可以使用isin()方法选出存在于列表 lst 中的数据行：

df[df['A'].isin(lst)]

输出：

        A   B
0   apple  10
1  banana  20

注意，isin() 方法返回一个布尔类型的 Series，其中 True 表示该项存在于 lst 中。因此我们也可以直接使用该方法获取布尔类型列表并进行筛选：

df[df['A'].isin(lst)]

输出：

array([ True, False,  True])

上面的结果表示，第一行和第三行数据存在于 lst 中。将该结果用作索引即可得到包含这些数据的 DataFrame。