使用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文件中的数据。
