在pandas中选择包含关键字的列名称

Question

我有这样的数据名人

d = {'col1': ['a', '2/1'], 'col2': ['b', 'c']}
df = pd.DataFrame(data=d)

我想知道哪些列包含键'/' 因为在列“col1”中，第一行包含键char“/”，所以我希望函数可以返回“col1”。在

Answer 1

您可以将列表理解与pd.Series.str.contains一起使用。在

res = [col for col in df.select_dtypes(include=[object]) if
       df[col].str.contains(r'/').any()]

print(res)  # ['col1']

在这个例子中，我显式地隔离了object类型的列，因为numeric/date/etc系列将超出您的测试范围。在