Python中使用csv模块读取指定列的数据

第一种方法，使用next()获取列名，然后使用index定位列名

import csv
with open(r'file.csv',encoding='utf-8',mode='r+') as file_read:
    csvread = csv.reader(file_read)
    headers = next(csvread) #获取csv文件中的列名
    content = headers.index('content') #提取csv文件中的指定列名
    # print(headers) #可打印列名查看csv文件中的columns。
    for row in csvread: #打印指定列的内容
        print(row[content])
file_read.close()

另一种方法是使用csv.DictReader()函数，将csv文件的每一行读成字典，代码更为简洁。

import csv
with open(r'file.csv',encoding='utf-8',mode='r+') as file_read:
    csvread = csv.DictReader(file_read)
    for row in csvread: #打印指定列的内容
        print(row['需要读取的该列列名'])
file_read.close()