使用pandas库对csv文件进行筛选保存


Posted in Python onMay 25, 2020

这个操作现在看来真没啥难的,但是我找相关的资料真的找了好久。

多数大佬都是直接pandas官网甩我脸上,然后举一个入门级的例子。

https://pandas.pydata.org/docs/reference/index.html

首先导入pandas库

import pandas as pd

然后使用read_csv来打开指定的csv文件

df = pd.read_csv('./IP2LOCATION.csv',encoding= 'utf-8')

这个函数里面需要写入csv文件的路径,如果是把csv文件保存到了python的工程文件夹下,则只需要./文件名即可,然后encoding='utf-8'是使用utf-8方式编码,有时候需要换成gbk。

虽然我们读取的是csv文件,但其实由于我们使用的是pandas库,所以我们实际获得的是一个DataFrame的数据结构。

可以使用print(type(df))进行检验

print(type(df))

使用pandas库对csv文件进行筛选保存

DataFrame 是表格型的数据结构。因此,我们可以将其当做表格。DataFrame 是以表格类似展示,而且还包含行标签、列标签。

我们可以添加一个列标签,使用方法为pandas.DataFrame.columns

在我们的例子中DataFrame类型的变量为df,因此使用方法为df.columns,我们添加的列标签为a、b、c、d、e、f

df.columns = ['a','b','c','d','e','f']

然后,我们想把某一列中等于特定值的那些行提取出来

可以将读出来的内容当做一个列表,然后这个列表的元素是表中的每一行,然后这每一行也是一个列表,也就是列表中的列表。

比如,我想将表中第5列中值为Andhra Pradesh的行提取出来,并且由于我们之前定义了第五列的列标签为e

因此代码为:

data = df[df['e'] == 'Andhra Pradesh']

最后我们可以通过pandas中的to_csv,来将筛选出来的数据保存到新的csv文件中。

data.to_csv('my_IP2LOCATION.csv')

用法为表名.to_csv('所要保存地方的路径/表名.csv')

最后总结一下我们的代码

import pandas as pd

df = pd.read_csv('./IP2LOCATION.csv',encoding= 'utf-8')
# print(type(df))
df.columns = ['a','b','c','d','e','f']
data = df[df['e'] == 'Andhra Pradesh']
data.to_csv('my_IP2LOCATION.csv')

IP2LOCATION.csv内容如下:

使用pandas库对csv文件进行筛选保存

总共有759727行

然后经过我们的筛选后的my_IP2LOCATION.csv

使用pandas库对csv文件进行筛选保存

只有3461行

PS:可以使用print(len(df.values))来查看行数

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
按日期打印Python的Tornado框架中的日志的方法
May 02 Python
Python获取任意xml节点值的方法
May 05 Python
详解Python中 __get__和__getattr__和__getattribute__的区别
Jun 16 Python
取numpy数组的某几行某几列方法
Apr 03 Python
Python中多个数组行合并及列合并的方法总结
Apr 12 Python
Python实现获取nginx服务器ip及流量统计信息功能示例
May 18 Python
python中实现字符串翻转的方法
Jul 11 Python
Python实现多进程的四种方式
Feb 22 Python
简单了解python单例模式的几种写法
Jul 01 Python
python设置环境变量的作用和实例
Jul 09 Python
python小程序实现刷票功能详解
Jul 17 Python
python爬虫中抓取指数的实例讲解
Dec 01 Python
pytorch中 gpu与gpu、gpu与cpu 在load时相互转化操作
May 25 #Python
基于pandas向csv添加新的行和列
May 25 #Python
Python如何把十进制数转换成ip地址
May 25 #Python
tensorflow模型转ncnn的操作方式
May 25 #Python
MxNet预训练模型到Pytorch模型的转换方式
May 25 #Python
浅谈pytorch 模型 .pt, .pth, .pkl的区别及模型保存方式
May 25 #Python
Pytorch通过保存为ONNX模型转TensorRT5的实现
May 25 #Python
You might like
php将mysql数据库整库导出生成sql文件的具体实现
2014/01/08 PHP
自定义min版smarty模板引擎MinSmarty.class.php文件及用法
2016/05/20 PHP
PHP实现的随机IP函数【国内IP段】
2016/07/20 PHP
页面只有一个text的时候,回车自动submit的解决方法
2010/08/12 Javascript
JS实现日期加减的方法
2013/11/29 Javascript
jquery实现的Accordion折叠面板效果代码
2015/09/02 Javascript
Webpack 实现 Node.js 代码热替换
2015/10/22 Javascript
JavaScript 消息框效果【实现代码】
2016/04/27 Javascript
js实现无缝滚动图(可控制当前滚动的方向)
2017/02/22 Javascript
vue.js事件处理器是什么
2017/03/20 Javascript
微信 jssdk 签名错误invalid signature的解决方法
2019/01/14 Javascript
vue实现菜单切换功能
2019/05/08 Javascript
JS中超越现实的匿名函数用法实例分析
2019/06/21 Javascript
Weex开发之WEEX-EROS开发踩坑(小结)
2019/10/16 Javascript
python获取网页状态码示例
2014/03/30 Python
Python 爬虫学习笔记之正则表达式
2016/09/21 Python
Python实现将16进制字符串转化为ascii字符的方法分析
2017/07/21 Python
深入理解Python爬虫代理池服务
2018/02/28 Python
python3连接MySQL数据库实例详解
2018/05/24 Python
Python 元组操作总结
2019/09/18 Python
pycharm 添加解释器的方法步骤
2020/08/31 Python
基于Python正确读取资源文件
2020/09/14 Python
使用python实现学生信息管理系统
2021/02/25 Python
瑞士灯具购物网站:Lampenwelt.ch
2018/07/08 全球购物
EJB的基本架构
2016/09/22 面试题
绘画专业自荐信范文
2014/02/23 职场文书
餐饮业员工工作决心书
2014/03/11 职场文书
保险公司晨会主持词
2014/03/22 职场文书
我的祖国演讲稿
2014/05/04 职场文书
小学班级特色活动方案
2014/08/31 职场文书
自习课吵闹检讨书范文
2014/09/26 职场文书
房屋租赁协议书(标准版)
2014/10/02 职场文书
简单的个人租房协议书范本
2014/11/26 职场文书
幼儿园小班教学反思
2016/03/03 职场文书
golang连接MySQl使用sqlx库
2022/04/14 Golang
python实现学员管理系统(面向对象版)
2022/06/05 Python