5.1 基本用法

# 读取 CSV 文件

with open('example.csv', 'r') as file:

reader = csv.reader(file)

for row in reader:

print(row)

5.2 高级用法

使用 DictReader 读取 CSV，每行数据被映射到一个字典。

with open('example.csv', 'r') as file:

reader = csv.DictReader(file)

for row in reader:

print(row)

6 实例演示

以下是一个 Python 示例，演示如何使用 csv 模块来读取和写入 CSV 文件。这个示例将包含两个主要部分：一个用于写入 CSV 文件的函数和另一个用于读取 CSV 文件的函数。我们将这些功能封装在一个名为 csv_example.py 的 Python 文件中。

# csv_example.py

import csv

def write_csv(file_name, data, field_names=None):

""" 写入 CSV 文件 """

with open(file_name, 'w', newline='', encoding='utf-8') as file:code>

if field_names:

writer = csv.DictWriter(file, fieldnames=field_names)

writer.writeheader()

writer.writerows(data)

else:

writer = csv.writer(file)

writer.writerows(data)

def read_csv(file_name):

""" 读取 CSV 文件 """

with open(file_name, 'r', newline='', encoding='utf-8') as file:code>

reader = csv.reader(file)

for row in reader:

print(row)

def main():

# CSV 文件名

file_name = 'example.csv'

# 数据 - 字典格式

data_dict = [

{ 'Name': 'Alice', 'Age': 30, 'City': 'New York'},

{ 'Name': 'Bob', 'Age': 25, 'City': 'Los Angeles'}

]

# 数据 - 列表格式

data_list = [

['Name', 'Age', 'City'],

['Alice', 30, 'New York'],

['Bob', 25, 'Los Angeles']

]

# 写入 CSV 文件

write_csv(file_name, data_dict, field_names=['Name', 'Age', 'City'])

# 或者使用列表数据

# write_csv(file_name, data_list)

# 读取 CSV 文件

read_csv(file_name)

if __name__ == '__main__':

main()

运行以上代码控制台输出结果：

[‘Name’, ‘Age’, ‘City’]

[‘Alice’, ‘30’, ‘New York’]

[‘Bob’, ‘25’, ‘Los Angeles’]

并生成文件：“example.csv”，内容如下：