Python如何读取和写入CSV文件
发布时间:2023-05-27 07:19:11
CSV文件是一种常用的数据交换格式,它的全称是Comma-Separated Values,即逗号分隔值文件,其数据以行为单位,每行数据通过逗号进行分隔。在Python中,我们可以使用csv模块进行CSV文件的读取和写入操作。
CSV文件的读取操作:
要读取CSV文件,我们需要先导入csv模块,然后使用csv.reader()函数读取文件,代码如下:
import csv
with open('data.csv') as csv_file:
csv_reader = csv.reader(csv_file)
for row in csv_reader:
print(row)
在上述代码中,我们首先打开一个名为“data.csv”的CSV文件,然后使用csv.reader()函数读取文件,并将其存储在变量csv_reader中。最后,我们使用一个循环来遍历所有行,并将它们打印出来。
如果CSV文件的 行包含列名,我们可以使用next()函数跳过该行,这样就可以忽略这些列名了,代码如下:
with open('data.csv') as csv_file:
csv_reader = csv.reader(csv_file)
next(csv_reader) # skip header row
for row in csv_reader:
print(row)
CSV文件的写入操作:
与读取CSV文件不同,写入CSV文件需要使用csv.writer()函数,并指定文件是否为二进制模式。代码如下:
import csv
with open('data.csv', mode='w', newline='') as csv_file:
csv_writer = csv.writer(csv_file)
csv_writer.writerow(['Name', 'Age', 'City'])
csv_writer.writerow(['Mary', '25', 'London'])
csv_writer.writerow(['John', '30', 'Paris'])
在上述代码中,我们首先打开一个名为“data.csv”的CSV文件,指定它是以写入模式打开的,并将newline参数设置为空字符串,这样可以避免写入空行。然后我们使用csv.writer()函数创建一个csv_writer对象,用于将数据写入文件。接着,我们使用writerow()函数写入表格的 行和其他行。
总结:
本文讨论了如何使用Python中的csv模块读取和写入CSV文件。初学者可能需要花费一些时间来熟悉这些操作,但一旦你掌握了这些技能,将能够更容易地导入和导出大量数据,这对数据分析和数据处理等任务非常有帮助。
