欢迎访问宙启技术站
智能推送

使用Python读取和写入CSV文件

发布时间:2023-12-04 11:37:41

CSV(Comma-Separated Values)是一种常用的纯文本文件格式,用于存储和交换简单的表格数据。Python提供了多种方法来读取和写入CSV文件。

读取CSV文件:

在Python中,我们可以使用csv模块来读取CSV文件。首先,我们需要导入csv模块:

import csv

接下来,我们可以使用open函数打开一个CSV文件,并将文件句柄传递给csv.reader函数来创建一个CSV阅读器对象。然后,我们可以使用for循环遍历阅读器对象,逐行读取CSV文件的内容。

以下是一个读取CSV文件的例子:

import csv

with open('data.csv', 'r') as file:
    reader = csv.reader(file)
    for row in reader:
        print(row)

上述代码打开名为data.csv的CSV文件,并创建一个CSV阅读器对象reader。然后,通过for循环遍历reader,逐行打印CSV文件的内容。

写入CSV文件:

与读取CSV文件类似,我们可以使用csv模块来写入CSV文件。我们需要引入csv模块,打开一个CSV文件,并将文件句柄传递给csv.writer函数来创建一个CSV写入器对象。然后,我们可以使用CSV写入器对象的writerow方法来将数据写入CSV文件。

以下是一个写入CSV文件的例子:

import csv

data = [
    ['Name', 'Age', 'Country'],
    ['John', '25', 'USA'],
    ['Emily', '30', 'Canada'],
    ['Tom', '35', 'UK']
]

with open('data.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(data)

在上述代码中,我们创建一个名为data的列表,其中包含每行数据的列表。然后,我们使用open函数打开名为data.csv的CSV文件,并创建一个CSV写入器对象writer。接下来,我们使用writer.writerows方法将数据写入CSV文件。

注意事项:

- 打开CSV文件时,需要指定打开模式(通常是'r'用于读取和'w'用于写入)。

- 读取CSV文件时,默认情况下,csv.reader将数据解析为字符串。如果需要将数据解析为其他数据类型,请使用适当的方法进行转换。

- 写入CSV文件时,请确保提供正确的行结束符(newline=''),以避免在Windows系统上出现空行。

总结:

Python的csv模块提供了方便的方法来读取和写入CSV文件。通过使用csv.reader函数和csv.writer函数,我们可以轻松地处理CSV文件中的数据。读取和写入CSV文件是处理表格数据的一种常见任务,Python的csv模块使这些任务变得简单而高效。无论是处理大量的数据,还是进行快速的数据分析,Python的csv模块都能满足我们的需求。