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

使用Python函数来读取和写入CSV文件

发布时间:2023-07-04 01:52:08

在Python中,我们可以使用内置的csv模块来读取和写入CSV文件。

首先,我们需要导入csv模块:

import csv

读取CSV文件:

首先,我们需要打开CSV文件并创建一个csv.reader对象,然后使用for循环逐行读取文件内容。例如,假设我们有一个名为"data.csv"的CSV文件,它包含以下内容:

Name, Age, City
John, 25, New York
Sarah, 30, London
Tom, 35, Paris

我们可以使用下面的代码来读取该文件:

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

上述代码将输出:

['Name', 'Age', 'City']
['John', '25', 'New York']
['Sarah', '30', 'London']
['Tom', '35', 'Paris']

写入CSV文件:

与读取CSV文件类似,我们需要打开一个CSV文件并创建一个csv.writer对象来写入数据。然后,我们可以使用writerow()方法来写入一行数据。例如,我们想将以下数据写入名为"output.csv"的CSV文件中:

Name, Age, City
Mike, 40, Berlin

我们可以使用下面的代码来实现:

data = [
    ['Name', 'Age', 'City'],
    ['Mike', '40', 'Berlin']
]

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

上述代码将创建一个名为"output.csv"的文件,并将data中的数据写入该文件。

除了使用writerows()方法写入多行数据外,我们还可以使用writerow()方法逐行写入数据。例如,我们可以使用下面的代码来写入多行数据:

with open('output.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    
    writer.writerow(['Name', 'Age', 'City'])
    writer.writerow(['Mike', '40', 'Berlin'])

上述代码将得到相同的结果。

当写入数据时,请确保将文件打开模式设置为'w',并将newline参数设置为空字符串,以避免在CSV文件中出现空行。

这就是如何使用Python函数来读取和写入CSV文件。使用csv模块,我们可以轻松地处理CSV文件中的数据。