pandas筛选出含有特定字符串的列

最新推荐文章于 2024-03-25 16:40:53 发布

SUFEHeisenberg

最新推荐文章于 2024-03-25 16:40:53 发布

阅读量809

点赞数

分类专栏： Data Processing 文章标签： pandas python 数据分析

本文链接：https://blog.csdn.net/weixin_43557139/article/details/127781053

版权

Data Processing 专栏收录该内容

22 篇文章 0 订阅

订阅专栏

本文介绍了如何使用Pandas库在Python中筛选出包含特定字符串的列，并展示了对筛选后的列进行求和与计算均值的操作。通过`df.columns.str.contains('col')`可以筛选出含有'col'的列，然后利用`.loc`选择这些列，进一步进行统计分析。

摘要由CSDN通过智能技术生成

pandas筛选出含有特定字符串的列

@author: Heisenberg

@date: 2022/11/08

df = df = pd.DataFrame(np.random.randn(-1,1,size=(5, 6)), columns=['A','B','C','A_col','B_col','C_col'])

#   A  B  C  A_col  B_col  C_col
#0  0 -1 -1      0      0     -1
#1  0 -1 -1      0     -1      0
#2  0  0 -1     -1      0     -1
#3  0  0  0      0      0      0
#4  0 -1 -1     -1     -1      0

df.loc[:,df.columns.str.contains('col')]
#   A_col  B_col  C_col
#0      0      0     -1
#1      0     -1      0
#2     -1      0     -1
#3      0      0      0
#4     -1     -1      0

#在此基础上就可以做很多操作了，比如这些特征列或者行的和、列/行的均值等。
mean_col = df.loc[:,df.columns.str.contains('col')].mean(axis=0)
sum_col = df.loc[:,df.columns.str.contains('col')].sum(axis=0)