python读取csv文件

在python里面，读取或写入csv文件时，首先要import csv这个库，然后利用这个库提供的方法进行对文件的读写。

 

0x01：获取每一行

读取csv文件，用的是csv.reader()这个方法。返回结果是一个_csv.reader的对象，我们可以对这个对象进行遍历，输出每一行，某一行，或某一列。代码如下：

import csv
with open('data.csv', 'r') as f:
    reader = csv.reader(f)
    print(type(reader))
   
    for row in reader:
        print(row)

以列表的形式输出每一行，如下：

0x02：获取某一行

如果你只想要其中的一行，可以先对reader进行一个类型转换，用list函数把它转换成列表，然后对对列表进行取元素，就可以获取到某一行的内容了，如下：

import csv
with open('data.csv', 'r') as f:
    reader = csv.reader(f)
    result = list(reader)
    print(result[1])

这里，我们获取的是第二行的内容，与上面的第二行的内容相同。

0x03：获取某一列

如果想要获取到其中一列的内容，可以在第一种情况下，对输出加一个下标，这样输出的就是某一列的内容的。

import csv
with open('data.csv', 'r') as f:
    reader = csv.reader(f)
    for i in reader:
        print(i[0])

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