Выберите имена столбцов, содержащие ключевое слово в пандах

Question

У меня есть подобное datafame

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, так как ряд числовых / даты / и т. Д. Будет выходить за рамкиваш тест.