python - 根据关键词 Pandas DF 提取列名

Question

我有一个像这样的数据框df，我想从数据框中提取列名称。这是我的 df。

index my_numbers  nums
0     648702831   9
1      26533315   8
2           366   3
3     354701058   9
4      25708239   8
5         70554   5
6       1574512   7
7          3975   4

我需要检查是否存在 numbers 或 num ，然后我想提取所有这些列。

我做了以下事情，

needed_cols = []
for i in df.columns:
    if 'numbers' or 'num' in i:
        needed_cols.append(i)

我得到以下结果，

['index', 'my_numbers', 'nums']

我只需要这些列，

['my_numbers', 'nums']

我怎样才能得到它们？想知道为什么我的代码不起作用。

Answer 1

因为 'num' 始终位于 'numbers' 中，您可以检查列是否包含 'num' 吗？

例如:

df.columns[df.columns.str.contains('num')]

Index(['my_numbers', 'nums'], dtype='object')