返回
Python CSV文件创建指南:快速上手并应用于数据存储
后端
2023-11-24 05:29:41
Python CSV 文件创建详解
CSV 文件,即逗号分隔值文件,是一种简单却强大的文本文件格式,广泛应用于数据存储和交换。由于其简洁的结构和广泛的兼容性,Python 语言内置了对 CSV 文件的操作支持,使其成为数据处理和分析的利器。
导入 CSV 模块
在 Python 中,处理 CSV 文件需要导入标准库中的 csv 模块。该模块提供了丰富的函数和方法,可以轻松创建、读取和修改 CSV 文件。
import csv
创建一个 CSV 文件
创建一个新的 CSV 文件非常简单,可以使用 open() 函数以写模式打开一个文件,并使用 csv.writer() 函数创建一个 CSV 写入对象。
with open('data.csv', 'w', newline='') as csvfile:
csvwriter = csv.writer(csvfile)
这里,我们以写模式打开了一个名为 data.csv 的新文件,并使用 newline='' 参数来防止在写入时添加额外的空行。
将数据写入 CSV 文件
使用 csvwriter.writerow() 方法可以将数据写入 CSV 文件。该方法接受一个列表作为参数,其中包含要写入的每一列数据。
csvwriter.writerow(['Name', 'Age', 'City'])
csvwriter.writerow(['John', '25', 'New York'])
csvwriter.writerow(['Mary', '30', 'London'])
在上面的代码中,我们首先写入了一行标题,然后写入了两行数据。
关闭 CSV 文件
在完成写入操作后,需要关闭 CSV 文件。
csvfile.close()
读取 CSV 文件
要读取 CSV 文件,可以使用 open() 函数以读模式打开文件,然后使用 csv.reader() 函数创建一个 CSV 读取对象。
with open('data.csv', 'r') as csvfile:
csvreader = csv.reader(csvfile)
这里,我们以读模式打开了一个名为 data.csv 的文件,并使用 csvreader 对象来读取文件中的数据。
从 CSV 文件中读取数据
使用 csvreader.next() 方法可以逐行读取 CSV 文件中的数据。
for row in csvreader:
print(row)
在上面的代码中,我们将每一行数据打印到了屏幕上。
总结
本指南介绍了如何使用 Python 创建、写入和读取 CSV 文件。这些技能在数据处理和分析中非常有用。