在Python中从CSV文件访问元素

什么是CSV文件

CSV指的是Comma-Separated Values(逗号分隔值)，其实就是一种用逗号分隔字段的纯文本文件格式。CSV文件的结构非常简单，常常用于数据的导入和导出。

以以下示例为例，是一个包含5个数据字段的CSV文件：

姓名,年龄,性别,职业,收入
Tom,28,男,工程师,8000
Lucy,24,女,设计师,6000
Peter,32,男,经理,12000
Lily,23,女,行政助理,4000

如何在Python中访问CSV文件中的元素

Python中有一个标准库csv，可以方便的对CSV文件进行读写操作。

读取CSV文件中的元素

可以使用csv模块中的reader()函数逐行读取CSV文件中的数据：

import csv

with open('data.csv') as csvfile:
    reader = csv.reader(csvfile)
    for row in reader:
        print(row)

输出结果如下：

['姓名', '年龄', '性别', '职业', '收入']
['Tom', '28', '男', '工程师', '8000']
['Lucy', '24', '女', '设计师', '6000']
['Peter', '32', '男', '经理', '12000']
['Lily', '23', '女', '行政助理', '4000']

也可以使用csv模块中的DictReader()函数逐行读取CSV文件中的数据，返回一个字典，键为表头字段名，值为对应的字段值：

import csv

with open('data.csv') as csvfile:
    reader = csv.DictReader(csvfile)
    for row in reader:
        print(row)

输出结果如下：

{'姓名': 'Tom', '年龄': '28', '性别': '男', '职业': '工程师', '收入': '8000'}
{'姓名': 'Lucy', '年龄': '24', '性别': '女', '职业': '设计师', '收入': '6000'}
{'姓名': 'Peter', '年龄': '32', '性别': '男', '职业': '经理', '收入': '12000'}
{'姓名': 'Lily', '年龄': '23', '性别': '女', '职业': '行政助理', '收入': '4000'}

写入CSV文件

可以使用csv模块中的writer()函数写入CSV文件：

import csv

with open('data.csv', 'w') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerow(['姓名', '年龄', '性别', '职业', '收入'])
    writer.writerow(['Tom', '28', '男', '工程师', '8000'])
    writer.writerow(['Lucy', '24', '女', '设计师', '6000'])
    writer.writerow(['Peter', '32', '男', '经理', '12000'])
    writer.writerow(['Lily', '23', '女', '行政助理', '4000'])

这样就将数据写入到了data.csv文件中。

总结

以上就是Python中如何访问CSV文件中的元素，通过csv模块，我们可以方便地读写CSV文件。