首页 > 资讯 > 后端开发 > Python >pandas中提取DataFrame某些列的一些方法

846

分享到

pandas中提取DataFrame某些列的一些方法

2024-04-02 19:04:59 846人浏览薄情痞子

Python 官方文档：入门教程 => 点击学习

摘要

目录前言方法一：df[columns]方法二：df.loc[]：用 label （行名或列名）做索引。方法三：df.iloc[]: i 表示 integer，用 integer lo

前言
方法一：df[columns]
方法二：df.loc[]：用 label （行名或列名）做索引。
方法三：df.iloc[]: i 表示 integer，用 integer location（行或列的整数位置，从0开始）做索引。
补充：提取所有列名中包含“线索”、“浏览”字段的列
参考：
总结

前言

在处理表格型数据时，一行数据是一个 sample，列就是待提取的特征。怎么选取其中的一些列呢？本文分享一些方法。

使用如下的数据作为例子：

import pandas as pd

data = pd.DataFrame({'Name':['Anna', 'Betty', 'Richard', 'Philip','Paul'],
        'course1':[85,83,90,84,85],
        'course2':[90,85,83,88,84],
        'course3':[82,86,81,91,85],
        'fruit':['apple','banana','apple','orange','peach'],
        'sport':['basketball', 'volleyball', 'football', 'basketball','baseball']},
         index=[1,2,3,4,5])
  
df = pd.DataFrame(data)

df

	Name	course1	course2	course3	fruit	sport
1	Anna	85	90	82	apple	basketball
2	Betty	83	85	86	banana	volleyball
3	Richard	90	83	81	apple	football
4	Philip	84	88	91	orange	basketball
5	Paul	85	84	85	peach	baseball

方法一：df[columns]

先看最简单的情况。输入列名，选择一列。例如：

df['course2']

1    90
2    85
3    83
4    88
5    84
Name: course2, dtype: int64

df[column list]：选择列。例如：

df[['course2','fruit']]

	course2	fruit
1	90	apple
2	85	banana
3	83	apple
4	88	orange
5	84	peach

或者以 column list (list 变量)的形式导入到 df[ ] 中，例如：

select_cols=['course2','fruit']
df[select_cols]

	course2	fruit
1	90	apple
2	85	banana
3	83	apple
4	88	orange
5	84	peach

可以用 column list=df.columns[start:end] 的方式选择连续列，start 和 end 均为数字，不包括 end 列。例如：

select_cols=df.columns[1:4]
df[select_cols]

	course1	course2	course3
1	85	90	82
2	83	85	86
3	90	83	81
4	84	88	91
5	85	84	85

你可能注意到，其中有 3 列的名字相近：‘course1’,‘course2’,‘course3’。怎么提取这三列呢？这里分享在Kaggle 上看到一位大神使用的 list comprehension方法。

select_cols=[c for c in df.columns if 'course' in c]
df[select_cols]

	course1	course2	course3
1	85	90	82
2	83	85	86
3	90	83	81
4	84	88	91
5	85	84	85

但是，如果你想输入df['course1':'course3'] 来索引连续列，就会报错。而输入数字索引df[1:3]时，结果不再是列索引，而是行索引，如下所示：

df[1:3]

	Name	course1	course2	course3	fruit	sport
2	Betty	83	85	86	banana	volleyball
3	Richard	90	83	81	apple	football

以下两种方法 df.loc[]和df.iloc[]就可以解决这个问题，可以明确行或列索引。还可以同时取多行和多列。

方法二：df.loc[]：用 label （行名或列名）做索引。

输入 column_list 选择多列 [:, column_list]，括号中第一个: 表示选择全部行。例如：

df.loc[:,['course2','fruit']]

	course2	fruit
1	90	apple
2	85	banana
3	83	apple
4	88	orange
5	84	peach

选择连续多列 [:,start_col: end_col]，注意：包括 end_col。例如：

df.loc[:,'course2':'fruit']

	course2	course3	fruit
1	90	82	apple
2	85	86	banana
3	83	81	apple
4	88	91	orange
5	84	85	peach

选择多行和多列，例如：

df.loc[1:3,'course2':'fruit']

	course2	course3	fruit
1	90	82	apple
2	85	86	banana
3	83	81	apple

与 df[ ]类似，df.loc[ ]括号内也可以输入判断语句，结果是对行做筛选。例如：

df.loc[df['course1']>84]
#注：输入df[df['course1']>84]，输出结果相同

	Name	course1	course2	course3	fruit	sport
1	Anna	85	90	82	apple	basketball
3	Richard	90	83	81	apple	football
5	Paul	85	84	85	peach	baseball

方法三：df.iloc[]: i 表示 integer，用 integer location（行或列的整数位置，从0开始）做索引。

df.iloc与df.loc用法类似，只是索引项不同。

df.iloc[:,[2,4]]

	course2	fruit
1	90	apple
2	85	banana
3	83	apple
4	88	orange
5	84	peach

选择连续多列：df.iloc[:, start_ix:end_ix]，注意：不包括 end_ix。例如：

df.iloc[:,2:5]

	course2	course3	fruit
1	90	82	apple
2	85	86	banana
3	83	81	apple
4	88	91	orange
5	84	85	peach

选择多行与多列，例如：

df.iloc[1:3,[2,4]]

	course2	fruit
2	85	banana
3	83	apple

与 df.loc[] 不同，df.iloc[] 括号内不可以输入判断语句。

补充：提取所有列名中包含“线索”、“浏览”字段的列

import pandas as pd
 
path = 'F:\python_projects\Python_learning\ershouche.csv'
df = pd.read_csv(open(path), index_col=0)
df = df.fillna(0)  # 填充空值后需赋值
print(df.describe())
 
columns = df.columns.values.tolist()  # 获取列名列表，注意values，tolist的使用
col_xian = []  # 存储包含‘线索'字段的列名
for i in columns:
    if '线索' in i:
        col_xian.append(i)
 
col_liu = [] # 存储包含‘浏览'字段的列名
for i in columns:
    if '浏览' in i:
        col_liu.append(i)
 
df_xian = df[col_xian]  # 根据列名取列
df_liu = df[col_liu]

参考：

1.如何选取dataframe的多列-教程：https://www.geeksforgeeks.org/how-to-select-multiple-columns-in-a-pandas-dataframe/

2.用 list comprehension 选择多列：Https://www.kaggle.com/code/robikscube/ieee-fraud-detection-first-look-and-eda/notebook

3.df.loc 与 df.iloc 的比较：https://stackoverflow.com/questions/31593201/how-are-iloc-and-loc-different

总结

到此这篇关于pandas中提取DataFrame某些列的文章就介绍到这了,更多相关pandas提取DataFrame某些列内容请搜索编程网以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程网！

您可能感兴趣的文档:

点击免费下载>>软考高级考试备考技巧/历年真题/备考精华资料

--结束END--

本文标题: pandas中提取DataFrame某些列的一些方法

本文链接: https://www.lsjlt.com/news/118520.html(转载时请注明来源链接)

有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

本篇文章演示代码以及资料文档资料下载

下载Word文档到电脑，方便收藏和打印～

下载Word文档

去做题

猜你喜欢

pandas中提取DataFrame某些列的一些方法

目录前言方法一：df[columns]方法二：df.loc[]：用 label （行名或列名）做索引。方法三：df.iloc[]: i 表示 integer，用 integer lo...

99+

2022-11-11
Pandas:如何修改DataFrame中某一列的值？

写这篇博客主要是因为在修改DataFrame列值的时候经常遇到bug，但到目前还没把这种错误复现出来。 DataFrame是Pandas中的主要数据结构之一，本篇博客主要介绍如何DataFr...

99+

2023-09-01

pandas python 数据分析
使用Pandas修改DataFrame中某一列的值

目录1 常规方法2. replace方法补充：DataFrame修改某一行某一列的值[坑点]总结写这篇博客主要是因为在修改DataFrame列值的时候经常遇到bug，但到...

99+

2022-12-22

修改dataframe中的某一个值 python 修改dataframe中某个值 pandas 修改某列数据
pandas取dataframe特定行列的实现方法

1.按列取、按索引/行取、按特定行列取 import numpy as np from pandas import DataFrame import pandas as pd ...

99+

2022-11-12
在Pandas DataFrame中插入一列的方法实例

目录引言示例 1：插入新列作为第一列示例 2：插入新列作为中间列示例 3：插入新列作为最后一列补充：按条件选择分组分别赋值总结引言通常，您可能希望在 Pandas DataFram...

99+

2022-11-13
「Python｜Pandas｜场景案例」如何只保留DataFrame数据集的某些列（要保留的列不固定）

本文主要介绍在使用pandas进行数据分析过程中的数据预处理时，如果希望仅保留某些列的数据需要如何操作。同时介绍一些特殊情况，比如列是用变量存储；或者列是一个全集，处理的数据集中不一定包括列出的全...

99+

2023-10-26

python pandas 数据分析机器学习
如何使用pandas读取csv文件中的某一列数据

使用pandas读取csv文件中的某一列数据，可以这样做：先导入pandas模块：import pandas as pd使用pd.read_csv函数读取csv文件：df = pd.read_csv("文件名.csv")使用df["列...

99+

2023-08-31

pandas python 数据分析数据挖掘机器学习
python随机获取列表中某一元素的方法

目录1、利用Python中的random模块中的choice方法2、利用Python中的random模块中的randrange方法3、利用Python中的random模块中的rand...

99+

2023-02-16

python随机获取列表元素 python随机获取元素
pandas提升计算效率的一些方法汇总

前言 Pandas是为一次性处理整个行或列的矢量化操作而设计的，循环遍历每个单元格、行或列并不是它的设计用途。所以，在使用Pandas时，你应该考虑高度可并行化的矩阵运算。一、避...

99+

2022-11-12
python列表中常见的一些排序方法

目录1、冒泡排序法方法一：直接使用for循环方法二：使用while语句2、选择排序法方法一：remove和append同时使用方法二：pop和append同时使用3、list.sor...

99+

2022-11-11
win10系统提示某些设置隐藏的解决方法

这篇文章将为大家详细讲解有关win10系统提示某些设置隐藏的解决方法，小编觉得挺实用的，因此分享给大家做个参考，希望大家阅读完这篇文章后可以有所收获。win10系统提示某些设置隐藏的详细关闭方法/步骤：打开电脑中的windows设置界面，随...

99+

2023-06-27
Linux shell 之提取文件名和目录名的一些方法总结

很多时候在使用Linux的shell时，我们都需要对文件名或目录名进行处理，通常的操作是由路径中提取出文件名，从路径中提取出目录名，提取文件后缀名等等。例如，从路径/dir1/dir2/file.txt中提...

99+

2022-06-04

文件名目录方法
vba提取网页数据的方法有哪些

在VBA中，提取网页数据的方法有以下几种：1. 使用IE对象：使用Internet Explorer对象来打开网页，并使用其提供的方...

99+

2023-09-12

vba
Pandas中批量替换字符的方法有哪些

本篇内容介绍了“Pandas中批量替换字符的方法有哪些”的有关知识，在实际案例的操作过程中，不少人都会遇到这样的困境，接下来就让小编带领大家学习一下如何处理这些情况吧！希望大家仔细阅读，能够学有所成！一、前言前几天在Python最强王者群有...

99+

2023-06-29
Shell脚本一次读取文件中一行的方法有哪些

这篇文章主要讲解了“Shell脚本一次读取文件中一行的方法有哪些”，文中的讲解内容简单清晰，易于学习与理解，下面请大家跟着小编的思路慢慢深入，一起来研究和学习“Shell脚本一次读取文件中一行的方法有哪些”吧！写法一:代码如下:#!/bin...

99+

2023-06-09
在PHP中获取一些特殊时间节点的方法

这篇文章给大家分享的是有关在PHP中获取一些特殊时间节点的方法的内容。小编觉得挺实用的，因此分享给大家做个参考，一起跟随小编过来看看吧。1.获取当月第一天及最后一天代码：<php$BeginMonthDate=date('Y-...

99+

2023-06-14
JavaScript中取整的方法有哪些

这篇文章主要讲解了“JavaScript中取整的方法有哪些”，文中的讲解内容简单清晰，易于学习与理解，下面请大家跟着小编的思路慢慢深入，一起来研究和学习“JavaScript中取整的方法有哪些”吧！ ...

99+

2022-10-19
爬虫解析提取数据的方法有哪些

本篇文章为大家展示了爬虫解析提取数据的方法有哪些，内容简明扼要并且容易理解，绝对能使你眼前一亮，通过这篇文章的详细介绍希望你能有所收获。一、分析网页以经典的爬取豆瓣电影 Top250 信息为例。每条电影信息在 ol class ...

99+

2023-06-15
有哪些Shell中删除某些文件外所有文件的方法

本篇内容介绍了“有哪些Shell中删除某些文件外所有文件的方法”的有关知识，在实际案例的操作过程中，不少人都会遇到这样的困境，接下来就让小编带领大家学习一下如何处理这些情况吧！希望大家仔细阅读，能够学有所成！假设要删除 ~/Download...

99+

2023-06-09
php查找数组中某个值的方法有哪些

本教程操作环境：windows7系统、PHP7.1版、DELL G3电脑php查找数组中某个值的方法方法1：使用in_array()函数in_array() 函数搜索数组中是否存在指定的值。<php header("Cont...

99+

2020-10-24

php 数组查找数组值