pandas 只获取匹配列表的条目 - Python

如果你使用 pandas 处理数据，且想只获取匹配列表的条目，可以使用下面的方法:

import pandas as pd

# 创建示例 DataFrame
df = pd.DataFrame({'列1': ['a', 'b', 'c', 'd', 'e'], '列2': [1, 2, 3, 4, 5]})

# 创建匹配列表
match_list = ['a', 'c', 'e']

# 只获取匹配列表的条目
result_df = df[df['列1'].isin(match_list)]

# 显示结果
print(result_df)

以上代码使用了 isin() 函数，该函数返回一个布尔型 Series，表示 DataFrame 中的每个元素是否匹配列表。传递该 Series 到 DataFrame 中，只有在索引为 True 的行才会保留。

以上就是 pandas 只获取匹配列表的条目的方法。