返回

Python CSV文件创建指南:快速上手并应用于数据存储

后端

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 文件。这些技能在数据处理和分析中非常有用。