在使用数字列表搜索列后返回数据框中的所有行 - Python/Pandas

Question

我一直在尝试在数据框中搜索数字列表，每次在列中匹配一个数字时，我想返回整行并将其保存到新的数据框，然后保存到 Excel。

millreflist 是数字列表 - 可以是随机长度。

TUCABCP 是我正在搜索的数据框。

PO 是我要在其中搜索数字的列。

我已经使用 .loc 尝试了下面的代码，但是在打开新的 excel 文件时，我只得到了标题，没有行或数据。

millreflistlength = len(millreflist)

for i in range(millreflistlength): TUCABCP = TUCABCP.loc[TUCABCP['PO'] == millreflist[i]]

TUCABCP.to_excel("NEWBCP.xlsx", header=True, index=False)

我使用了以下问题作为参考，但它不包括您何时想使用数字列表进行搜索：Selecting rows from a Dataframe based on values in multiple columns in pandas

Answer 1

试试这样的：

## Get list, where each element is the index of a row which you want to keep
indexes = TUCABCP[TUCABCP['PO'].isin(millreflist)]
## Filter the original df to get just the rows with indexes in the list
df = TUCABCP[TUCABCP.index.isin(indexes)]