在python里面,读取或写入csv文件时,首先要import csv这个库,然后利用这个库提供的方法进行对文件的读写。
0x01:获取每一行
读取csv文件,用的是csv.reader()这个方法。返回结果是一个_csv.reader的对象,我们可以对这个对象进行遍历,输出每一行,某一行,或某一列。代码如下:
1 import csv 2 with open('data.csv', 'r') as f: 3 reader = csv.reader(f) 4 print(type(reader)) 5 6 for row in reader: 7 print(row)
以列表的形式输出每一行,如下:
0x02:获取某一行
如果你只想要其中的一行,可以先对reader进行一个类型转换,用list函数把它转换成列表,然后对对列表进行取元素,就可以获取到某一行的内容了,如下:
1 import csv 2 with open('data.csv', 'r') as f: 3 reader = csv.reader(f) 4 result = list(reader) 5 print(result[1])
这里,我们获取的是第二行的内容,与上面的第二行的内容相同。
0x03:获取某一列
如果想要获取到其中一列的内容,可以在第一种情况下,对输出加一个下标,这样输出的就是某一列的内容的。
1 import csv 2 with open('data.csv', 'r') as f: 3 reader = csv.reader(f) 4 for i in reader: 5 print(i[0])
这里,获取第一列的内容,与第一次输出结果相同