如何在包含具有特定子字符串的列的数据框字典中选择数据框

Question

我有一本数据框字典df_dict。然后我有一个子字符串“blue”。我想在我的数据框字典中识别数据框的名称，该字典中至少有一个列的名称包含子字符串“blue”。

我正在考虑尝试类似的方法：

for df in df_dict:
     if df.columns.contains('blue'):
          return df
     else:
          pass

但是，我不确定这里是否需要 for 循环。如何在我的数据框字典中找到我要查找的数据框的名称？

Answer 1

我认为循环对于迭代字典项是必要的：

out = {name: df for name, df in df_dict.items() if df.columns.str.contains('blue').any()}

或者：

out = {name: df for name, df in df_dict.items() if any('blue' in y for y in df.columns)}