在数据框中查找多个列表值

Question

我有一个list 名称为list_1，如下所示：

[["xx ",'yy']
[["gg ",'xx']
[["xx ",'gg']]

另外，我有一个像这样的df：

  column1  column2............ coulymn_N
0  xx        gg                 gg
1  gg        xx                 yy
2  xx        something else     nan

我想将list_1逐行搜索到df中并返回匹配项

例如，df 中第一行 list_1 的输出可能是

  column1  
0  xx 

还有

  column_N
1   yy

我遇到的问题是，在我的list 中，我有多个值（"xx ", 和"yy")，所以我很困惑。我不认为我可以使用isin 方法。

有什么想法吗？

Answer 1

假设你有一个类似的列表：

list_of_lists = [["xx", "yy"], ["gg", "xx"], ["xx", "gg"]]

还有一个像这样的熊猫数据框：

import pandas as pd

df = pd.DataFrame(
    {
        "column1": ["xx", "yy", "aa"],
        "column2": ["yy", "gg", "aa"],
        "column3": ["gg", "aa", "hh"],
    }
)

然后，您可以遍历 list_of_lists 中的每个 sub_list，并检查 sub_list 的每个项目是否存在于数据框的任何列中。在这种情况下，您可以打印结果：

for sub_list in list_of_lists:
    print("\n")
    print("Sub_list: ", sub_list)
    for item in sub_list:
        for col in df.columns:
            if item in df[col].values:
                print(df[col][df[col] == item].to_frame())

上面将打印以下内容：

Sub_list:  ['xx', 'yy']
  column1
0      xx
  column1
1      yy
  column2
0      yy


Sub_list:  ['gg', 'xx']
  column2
1      gg
  column3
0      gg
  column1
0      xx


Sub_list:  ['xx', 'gg']
  column1
0      xx
  column2
1      gg
  column3
0      gg