在Python中,csv
模块是用于读写CSV(逗号分隔值)文件的。这个模块提供了reader()
和writer()
等函数,用于处理CSV文件。
技术名称
csv
模块reader()
函数:用于读取CSV文件中的数据writer()
函数:用于将数据写入CSV文件
Python代码示例
读取CSV文件
import csv
# 打开CSV文件
with open('example.csv', 'r', newline='') as file:
reader = csv.reader(file)
# 遍历CSV文件的每一行
for row in reader:
print(row) # 打印整行数据,row是一个列表
写入CSV文件
import csv
# 定义要写入的数据
data = [
['Name', 'Age', 'City'],
['Alice', 25, 'New York'],
['Bob', 30, 'San Francisco'],
['Charlie', 35, 'Los Angeles']
]
# 打开CSV文件(如果不存在则创建)
with open('output.csv', 'w', newline='') as file:
writer = csv.writer(file)
# 写入数据到CSV文件
for row in data:
writer.writerow(row) # 写入整行数据
在上面的示例中,我们首先导入了csv
模块。然后,我们使用with
语句打开CSV文件,这样可以确保文件在使用完毕后被正确关闭。对于读取操作,我们使用csv.reader()
函数创建一个读取器对象,并使用for
循环遍历文件的每一行。对于写入操作,我们使用csv.writer()
函数创建一个写入器对象,并使用writerow()
方法将每行数据写入文件。注意,在打开文件时,我们使用了newline=''
参数,这是为了防止在Windows系统中出现额外的空行。