使用Python读写csv文件

当涉及到处理数据的时候，CSV文件通常是人们选择的首选。因为CSV文件很简单，易于阅读和编写。Python提供了丰富的库和函数来读写CSV文件。下面就是使用Python读写CSV文件的完整攻略。

什么是CSV文件

CSV是Comma Separated Values的缩写，也就是逗号分隔值。CSV文件是一种简单的文件格式，用来存储数据表格，可以使用Excel或纯文本编辑器打开。CSV文件中的每一行代表一条数据记录，每个值之间用逗号隔开。

读取CSV文件

使用Python的csv模块可以轻松地读取CSV文件。下面是读取CSV文件的基本步骤：
1. 导入csv模块
2. 使用open函数打开CSV文件
3. 创建csv读取器(csv.reader)对象
4. 循环遍历读取器对象，处理每一行数据

示例一：读取CSV文件中的数据

import csv

# 打开CSV文件
with open('data.csv', 'r') as file:
    # 创建csv读取器对象
    reader = csv.reader(file)
    # 循环遍历读取器对象，处理每一行数据
    for row in reader:
        print(row)

代码中的csv.reader函数将CSV文件返回一个读取器对象，遍历读取器对象就可以循环遍历每一行数据。该代码将会输出CSV文件中的所有行。

示例二：读取CSV文件中特定的列

import csv

# 打开CSV文件
with open('data.csv', 'r') as file:
    # 创建csv读取器对象
    reader = csv.reader(file)
    # 循环遍历读取器对象，处理每一行数据
    for row in reader:
        # 输出特定的列
        print(row[0], row[-1])

代码中的row[0]表示输出每一行的第一个值，而row[-1]表示输出每一行的最后一个值。

写入CSV文件

使用Python的csv模块也可以实现将数据写入CSV文件。下面是将数据写入CSV文件的基本步骤：
1. 导入csv模块
2. 使用open函数打开CSV文件
3. 创建csv写入器(csv.writer)对象
4. 循环遍历数据，将数据写入CSV文件

示例三：写入CSV文件

import csv

# 假设的数据
data = [
    ['Tom', 23, 'China'],
    ['Lucy', 24, 'USA'],
    ['Alex', 21 , 'Canada']
]

# 打开CSV文件
with open('output.csv', 'w', newline='') as file:
    # 创建csv写入器对象
    writer = csv.writer(file)
    # 循环遍历每一行数据，将数据写入CSV文件
    for row in data:
        writer.writerow(row)

print('写入CSV文件成功！')

代码中的newline=''参数用来避免写入空行的问题。该代码将会输出一个output.csv文件，其中包含了data所定义的数据。