python读取csv文件

在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])

这里,获取第一列的内容,与第一次输出结果相同

原文地址:https://www.cnblogs.com/liangshian/p/11272155.html