python - 如果任何行包含特定字符串，则选择列

Question

如果列中的任何值包含字符串，我将尝试获取 DataFrame 中的列列表。例如，在下面的数据框中，我想要一个字符串中包含 % 的列列表。我能够使用 for 循环和 series.str.contains 方法来完成此操作，但似乎不是最佳选择，尤其是对于较大的数据集。有没有更有效的方法来做到这一点？

import pandas as pd

df = pd.DataFrame({'A': {0: '2019-06-01', 1: '2019-06-01', 2: '2019-06-01'},
                   'B': {0: '10', 1: '20', 2: '30'},
                   'C': {0: '10', 1: '20%', 2: '30%'},
                   'D': {0: '10%', 1: '20%', 2: '30'},
               })

数据框

            A   B    C    D
0  2019-06-01  10   10  10%
1  2019-06-01  20  20%  20%
2  2019-06-01  30  30%   30

当前方法

col_list = []
for col in df.columns:
    if (True in list(df[col].str.contains('%'))) is True:
        col_list.append(col)

输出

['C', 'D']

Answer 1

stack 与 any

df.columns[df.stack().str.contains('%').any(level=1)]

Index(['C', 'D'], dtype='object')

---

理解

[c for c in df if df[c].str.contains('%').any()]

['C', 'D']

---

过滤器

[*filter(lambda c: df[c].str.contains('%').any(), df)]

['C', 'D']

---

Numpy 的查找

from numpy.core.defchararray import find

df.columns[(find(df.to_numpy().astype(str), '%') >= 0).any(0)]

Index(['C', 'D'], dtype='object')