如何通过部分标签过滤pandas数据帧列

时间:2014-03-08 16:43:54

标签: python pandas dataframe

我正在尝试通过部分标签过滤pandas数据帧列(类型为pandas.core.index.Index)。

我正在寻找一种内置方法,它可以获得与以下相同的结果:

partial_label = 'partial_lab'
columns = df.columns
columns = [c for c in columns if c.startswith(partial_label)]
df = df[columns]

有没有内置的东西可以获得这个?

由于

1 个答案:

答案 0 :(得分:3)

可能的解决方案:

df.filter(regex='partial_lab.*')

idx = df.columns.to_series().str.startswith('partial_lab')
df.loc[:,idx]