python - 根据列名创建 DataFrame 的子集

Question

我有一个名为 timedata 的 pandas DataFrame，它具有不同的列名称，其中一些包含单词 Vibration，一些包含 eccentricity。是否可以创建仅包含振动一词的列的数据框？

我试过用

vib=[]
for i in timedata:
    if 'Vibration' in i:
        vib=vib.append(i)

然后根据这些列的索引创建一个 DataFrame。这看起来确实不是最有效的方法，我敢肯定列表理解一定有一些简单的方法。

编辑

表单数据框:

df = DataFrame({'Ch 1:Load': randn(10), 'Ch 2:Vibration Brg 1T ': randn(10), 'Ch 3:Eccentricity Brg 1H ': randn(10), 'Ch 4:Vibration Brg 2T ': randn(10)})

对不起，我今天过得很慢!感谢您的帮助

Answer 1

像这样手动选择其中包含“振动”一词的所有列:

df[[col for col in df.columns if "Vibration" in col]]

你也可以用 filter 方法做同样的事情:

df.filter(like="Vibration")

如果你想做一个更灵活的过滤器，你可以使用regex选项。例如。查看列名称中是否包含“Vibration”或“Ecc”:

df.filter(regex='Ecc|Vibration')