代码之家 › 专栏 › 技术社区 › Mooni

Python:从dataframe中选择多个相似的列

multiple-columns dataframe pandas python

Mooni · 技术社区 · 7 年前

我有一个数据帧:

df:

    a21   b21   c21  a22   b22  a23  b23
1    2    2      2    4     5    7    7
2    2    2      2    4     5    7    7
3    2    2      2    4     5    7    7
4    2    2      2    4     5    7    7
5    2    2      2    4     5    7    7

我只想选择具有 '21' '23' 这样我的输出是:

df_output:

    a21   b21   c21   a23  b23
1    2    2      2     7    7
2    2    2      2     7    7
3    2    2      2     7    7
4    2    2      2     7    7
5    2    2      2     7    7

我可以用以下代码来实现这一点:

df_21 = (df.loc[:, df.filter(like='21').columns])    
df_23 = (df.loc[:, df.filter(like='23').columns])

然后我可以合并 df_21 df_23

2 回复 | 直到 7 年前

MaxU - stand with Ukraine 7 年前

我们可以使用 DataFrame.filter() 方法:

In [38]: df.filter(regex=r'21|23')
Out[38]:
   a21  b21  c21  a23  b23
1    2    2    2    7    7
2    2    2    2    7    7
3    2    2    2    7    7
4    2    2    2    7    7
5    2    2    2    7    7

In [45]: df.loc[:, df.columns.str.contains(r'21|23')]
Out[45]:
   a21  b21  c21  a23  b23
1    2    2    2    7    7
2    2    2    2    7    7
3    2    2    2    7    7
4    2    2    2    7    7
5    2    2    2    7    7

Alexander 7 年前

您可以使用条件列表理解:

targets = ['21', '23']
df[[col for col in df if any(target in col for target in targets)]]

推荐文章

July · 如何定义数字间隔,然后四舍五入

1 年前

Community wiki · 对象名称前的单下划线和双下划线的含义是什么?

1 年前

Brian Johnson · 为什么在Python中列出字典列表会引发TypeError?[已关闭]

1 年前

user026 · 如何根据特定窗口的平均值(行数)创建新列?

1 年前

Ashok Shrestha · 需要追踪特定的颜色线并获取坐标

1 年前

Nicote Ool · 在FastApi和Vue3中获得422

1 年前

NeoExceptCodeBad · 如果我有很多垂直线,我如何找到它们的边缘?

1 年前

Abdulaziz · 如何对集合内的列表进行排序[重复]

1 年前

user2743931 · 带有src目录的Python setup.py

1 年前

asmgx · 为什么合并数据帧不能按照python中的预期方式工作

1 年前