python - 过滤掉 Pandas 数据框中的某些列

Question

我正在尝试根据列名过滤掉 pandas 数据框中的某些列。

所以框架大致是:

import pandas as pd

dict_build = {'Var_1_Reading': [1,2,3,10,0.1], 'Var_1_Confidence':[1,1,1,1,1], 'Var_2_Reading': [1,2,3,10,0.1], 'Var_2_Confidence':[1,1,1,1,1]}
df = pd.DataFrame(dict_build)

我想在末尾删除带有 Confidence 的列。我尝试使用 filter 但无法解决。目前我在做:

ColList_to_drop = df.filter(regex='Confidence', axis=1).columns.values
filtered_df = df.drop(ColList_to_drop, axis = 1)

我可以只在正则表达式中完成这两个步骤吗？

Answer 1

应该这样做:

>>> cols = [col for col in df.columns if not col.endswith('Confidence')]    
>>> df = df[cols]

>>> df
   Var_1_Reading  Var_2_Reading
0            1.0            1.0
1            2.0            2.0
2            3.0            3.0
3           10.0           10.0
4            0.1            0.1