准备数据
name,age
小明,14
小刚,15
通常,我们用下面的代码读取csv
import csv
with open(r'C:\Users\zhangdongsheng\Desktop\data.csv', encoding='utf-8')as f:
reader = csv.reader(f)
headers = next(reader)
print(headers)
for row in reader:
print(row)
程序输出结果
['name', 'age']
['小明', '14']
['小刚', '15']
我们完全可以像读取txt文件那样去读取csv文件,但那样读取到的数据一行就是一个字符串,还需你自己进行一次split操作才能将数据分隔开,因此我建议你使用csv模块读取csv文件。
上面的读取方法,有一个让认感到难受的地方,每一行数据虽然都是以列表的形式返回,可如果你想获取一行中的某一列数据,就只能通过列表的索引,这样并不方便。
针对这个需求,可以使用namedtuple
import csv
from collections import namedtuple
with open(r'C:\Users\zhangdongsheng\Desktop\data.csv', encoding='utf-8')as f:
reader = csv.reader(f)
headers = next(reader)
Row = namedtuple('Row', headers)
for row in reader:
row = Row(*row)
print(row.name, row.age)
这样可以非常方便的获取一行数据中的某一列数据。