9.3 CSV文档的读写¶

在 Python 中，可以使用 csv 模块或 pandas 库来读取和保存 CSV（逗号分隔值）文件，常用于存储和传输结构化数据。

9.3.1 创建并保存为 CSV 文件¶

In [1]:
import csv

data = [
    ['姓名', '年龄', '性别'],
    ['张三', '20', '男'],
    ['李四', '22', '女'],
    ['王五', '21', '男']
]

with open('example.csv', 'w', newline='', encoding='utf-8') as f:
    writer = csv.writer(f)
    writer.writerows(data)

9.3.2 读取 CSV 文件¶

In [2]:
import csv

with open('example.csv', 'r', encoding='utf-8') as f:
    reader = csv.reader(f)
    for row in reader:
        print(row)

['姓名', '年龄', '性别']
['张三', '20', '男']
['李四', '22', '女']
['王五', '21', '男']
In [ ]:
with open('example.csv', 'r', encoding='utf-8') as f:
    reader = csv.reader(f)
    header = next(reader)
    for i, row in enumerate(reader):
        name, age, gender = row
        age = int(age)
        print(name, age, gender)

9.3.3 利用 pandas 读写 CSV 文件¶

pandas 是 Python 中用于数据分析的强大工具，支持快速读写 CSV 文件。

9.3.4 写入 CSV 文件¶

In [4]:
import pandas as pd

df = pd.DataFrame({
    '姓名': ['张三', '李四', '王五'],
    '年龄': [20, 22, 21],
    '性别': ['男', '女', '男']
})

df.to_csv('data.csv', index=False)

# with open('data.csv', 'r', encoding='utf-8') as f:
#     print(f.read())

9.3.5 读取 CSV 文件¶

In [5]:
import pandas as pd

data = pd.read_csv('data.csv')
print(data)

   姓名  年龄 性别
0  张三  20  男
1  李四  22  女
2  王五  21  男
In [ ]: