用python读写和处理csv文件

用python读写和处理csv文件

---

读取

1、安装pandas

这里我们使用pandas包来读取csv文件，pandas处理csv文件十分方便,是我认为是目前最方便的读取方式。

首先安装pandas

pip install pandas

---

2、读取实例

安装好了之后我们读取一个csv文件

import pandas as pd                         #导入pandas包
data = pd.read_csv("train.csv")           	#读取csv文件
#print(data)                                #打印所有文件

print (data.head(5))                        #打印前5行

print(data.columns) 						#返回全部列名
print(data.shape)							#f返回csv文件形状
print(data.loc[1:2])						#打印第1到2行
data.loc[2:4, ['PassengerId', 'Sex']]       #打印行中特定列

更多有关pandans读取csv文件的内容请参考
https://blog.csdn.net/liufang0001/article/details/77856255

---

写入

import csv
csvFile = open("csvData.csv", "w")            #创建csv文件
writer = csv.writer(csvFile)                  #创建写的对象
#先写入columns_name                             
writer.writerow(["index","a_name","b_name"])     #写入列的名称
#写入多行用writerows                                #写入多行
writer.writerows([[1,a,b],[2,c,d],[3,d,e]])
csvFile.close()

生成的csv文件如下图