python - 将行与python中的值列表匹配

Question

我有一个 23 列的数据框，其中一个名为 column_lei 的列包含各种公司的 LEI，我还有一个名为 lei_codes 的列表，其中包含我需要在数据框中找到的大量特定 LEI。

我如何通过数据框中 lei 列中的每一行运行列表，如果该列中的任何行与列表中的任何值匹配，则数据框中的整行将被挑选出来，并且放入一个新的数据框中。所以最后我有一个新的数据框，其中包含所有 23 列记录，其中 LEI 列与列表匹配。

标签： pythonpandas

Answer 1

您可以isin()在“column_lei”列上使用该功能

这是一个包含 3 列的示例：

import pandas as pd
import numpy as np

df = pd.DataFrame({"a": np.repeat("a", 5),
                   "column_lei": [1,2,3,4,5],
                   "b": np.repeat("b", 5)})
lei_codes = [1,3,5]
df_new = df[df.column_lei.isin(lei_codes)]
df_new