使用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二分法搜索算法实例分析
May 11 Python
Python实现快速排序算法及去重的快速排序的简单示例
Jun 26 Python
python xlsxwriter库生成图表的应用示例
Mar 16 Python
解决pip install xxx报错SyntaxError: invalid syntax的问题
Nov 30 Python
python实现简单多人聊天室
Dec 11 Python
python解析含有重复key的json方法
Jan 22 Python
Python实现定时执行任务的三种方式简单示例
Mar 30 Python
Flask框架模板渲染操作简单示例
Jul 31 Python
twilio python自动拨打电话,播放自定义mp3音频的方法
Aug 08 Python
pytorch方法测试——激活函数(ReLU)详解
Jan 15 Python
keras实现theano和tensorflow训练的模型相互转换
Jun 19 Python
通过实例解析Python文件操作实现步骤
Sep 21 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 多行多列显示
2009/08/15 PHP
PHP图片水印类的封装
2017/07/06 PHP
php使用fullcalendar日历插件详解
2019/03/06 PHP
php将字符串转换为数组实例讲解
2020/05/05 PHP
Javascript - HTML的request类
2007/01/09 Javascript
Javascript 汉字字节判断
2009/08/01 Javascript
IE与firefox下Dhtml的一些区别小结
2009/12/02 Javascript
基于jquery的高性能td和input切换并可修改内容实现代码
2011/01/09 Javascript
理解Javascript闭包
2013/11/01 Javascript
当json键为数字时的取值方法解析
2013/11/15 Javascript
使用jsonp完美解决跨域问题
2014/11/27 Javascript
jQuery实现动画效果circle实例
2015/08/06 Javascript
Bootstrap模态框调用功能实现方法
2016/09/19 Javascript
详解vue的数据binding绑定原理
2017/04/12 Javascript
webpack多页面开发实践
2017/12/18 Javascript
JS实现的定时器展示简单秒表、页面弹框及跳转操作完整示例
2020/01/26 Javascript
vue element table中自定义一些input的验证操作
2020/07/18 Javascript
跟老齐学Python之用while来循环
2014/10/02 Python
python在windows和linux下获得本机本地ip地址方法小结
2015/03/20 Python
python使用BeautifulSoup分页网页中超链接的方法
2015/04/04 Python
python用reduce和map把字符串转为数字的方法
2016/12/19 Python
Python获取某一天是星期几的方法示例
2017/01/17 Python
numpy数组拼接简单示例
2017/12/15 Python
python通过tcp发送xml报文的方法
2018/12/28 Python
Python高级特性 切片 迭代解析
2019/08/23 Python
python中的TCP(传输控制协议)用法实例分析
2019/11/15 Python
使用opencv将视频帧转成图片输出
2019/12/10 Python
matplotlib 三维图表绘制方法简介
2020/09/20 Python
Bata印度官网:源自欧洲舒适鞋履品牌
2020/01/30 全球购物
Android面试宝典
2013/08/06 面试题
毕业典礼主持词大全
2014/03/26 职场文书
庆国庆国旗下讲话稿2014
2014/09/21 职场文书
教师自查自纠工作情况报告
2014/10/29 职场文书
2015年宣传工作总结
2015/04/08 职场文书
《妈妈别哭,有我在》读后感3篇
2020/01/13 职场文书
matplotlib画混淆矩阵与正确率曲线的实例代码
2021/06/01 Python