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

如何使用Python函数读取和写入CSV文件?

发布时间:2023-05-20 18:17:47

CSV文件是指以逗号分隔的值文件,在数据分析领域中被广泛使用。Python是一门强大的编程语言,它有许多内置函数和第三方库,可帮助我们读取和写入CSV文件。本文将介绍如何使用Python函数读取和写入CSV文件。

读取CSV文件:

在Python中,我们可以使用内置的csv模块来读取CSV文件。这个模块提供了一个函数csv.reader(),它可以帮助我们逐行读取CSV文件,并将其转换为列表或元组。

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

import csv

with open('filename.csv', 'r') as csv_file:

    csv_reader = csv.reader(csv_file)

    for line in csv_reader:

        print(line)

在这个例子中,我们打开一个名为“filename.csv”的文件,并使用csv.reader()函数读取文件。然后,我们使用一个for循环来遍历文件的每一行,并打印出来。

写入CSV文件:

类似于读取CSV文件,Python中也有csv模块来写入CSV文件。使用csv.writer()函数,我们可以将列表或元组数据写入CSV文件中。

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

import csv

data = [['Alice', '25'], ['Bob', '30'], ['Charlie', '35']]

with open('filename.csv', 'w', newline='') as csv_file:

    csv_writer = csv.writer(csv_file)

    for line in data:

        csv_writer.writerow(line)

在这个例子中,我们定义了一个名为“data”的列表,其中包含三个子列表,每个子列表包含两个值。然后,我们打开一个名为“filename.csv”的文件,并使用csv.writer()函数将数据写入文件中。接下来,我们使用一个for循环遍历数据的每一行,并使用csv_writer.writerow()函数将每一行写入文件中。

总结:

在Python中,我们可以使用csv模块读取和写入CSV文件。使用csv.reader()函数读取文件,并使用csv.writer()函数写入文件。这是一种常见的数据交换格式,在数据分析,科学和工业界都得到了广泛的应用。