查找名称包含特定字符串的列

小编典典

查找名称包含特定字符串的列

python

我有一个带有列名称的数据框，我想找到一个包含特定字符串但与之不完全匹配的数据框。我在寻找'spike'列名喜欢'spike-2'，'hey spike'，'spiked-in'（该'spike'部分总是连续）。

我希望列名以字符串或变量的形式返回，因此我以后可以使用df['name']或df[name]照常访问列。我试图找到方法，但没有成功。有小费吗？

阅读 212

2021-01-20

共1个答案

小编典典

只需遍历DataFrame.columns，现在是一个示例，在该示例中，您将获得匹配的列名称列表：

import pandas as pd

data = {'spike-2': [1,2,3], 'hey spke': [4,5,6], 'spiked-in': [7,8,9], 'no': [10,11,12]}
df = pd.DataFrame(data)

spike_cols = [col for col in df.columns if 'spike' in col]
print(list(df.columns))
print(spike_cols)

输出：

['hey spke', 'no', 'spike-2', 'spiked-in']
['spike-2', 'spiked-in']

说明：

df.columns 返回列名列表
[col for col in df.columns if 'spike' in col]``df.columns使用变量遍历列表col并将其添加到结果列表（如果col包含）'spike'。此语法是列表理解。

如果只希望结果数据集的列匹配，则可以执行以下操作：

df2 = df.filter(regex='spike')
print(df2)

输出：

   spike-2  spiked-in
0        1          7
1        2          8
2        3          9

2021-01-20