Python 3.x 中如何使用csv模块读取和写入CSV文件
引言:CSV(Comma-Separated Values)是一种常见的文件格式,用于存储表格数据。在Python中,可以使用csv模块来处理CSV文件,该模块提供了读取和写入CSV文件的功能。本文将介绍如何使用csv模块在Python 3.x中读取和写入CSV文件,并提供代码示例。
一、使用csv模块读取CSV文件
首先,我们需要导入csv模块:
import csv
登录后复制登录后复制
接下来,我们可以使用csv.reader()函数来读取CSV文件。该函数需要传入一个文件对象,我们可以使用open()函数来打开一个CSV文件,并把文件对象作为参数传给csv.reader()函数。
假设我们有一个名为data.csv的CSV文件,该文件的内容如下:
name,age,city
John,25,New York
Peter,30,Chicago
Sarah,28,Los Angeles
登录后复制
下面是一个简单的代码示例,展示如何使用csv模块来读取CSV文件:
import csv
with open('data.csv', 'r') as file:
csv_reader = csv.reader(file)
for row in csv_reader:
print(row)
登录后复制
运行上述代码,将会输出CSV文件中的每一行数据。在这个示例中,我们使用了一个for循环来遍历csv_reader对象,以逐行打印CSV文件中的数据。
二、使用csv模块写入CSV文件
与读取CSV文件类似,我们也可以使用csv模块来写入CSV文件。同样需要导入csv模块:
import csv
登录后复制登录后复制
然后,我们可以使用csv.writer()函数来创建一个csv_writer对象,该对象可以将数据写入CSV文件。该函数需要传入一个文件对象,同样可以使用open()函数来打开一个CSV文件,并把文件对象作为参数传给csv.writer()函数。
下面是一个简单的代码示例,展示如何使用csv模块来写入CSV文件:
import csv
data = [
['name', 'age', 'city'],
['John', '25', 'New York'],
['Peter', '30', 'Chicago'],
['Sarah', '28', 'Los Angeles']
]
with open('output.csv', 'w', newline='') as file:
csv_writer = csv.writer(file)
csv_writer.writerows(data)
登录后复制
运行上述代码,将会在当前目录下创建一个名为output.csv的CSV文件,并将data中的数据写入到该文件中。
在代码示例中,我们首先定义了一个二维列表data,其中包含了CSV文件的每一行数据。然后,我们使用csv.writerows()函数将整个列表写入CSV文件。
总结:通过csv模块,我们可以很方便地在Python 3.x中读取和写入CSV文件。在读取CSV文件时,我们可以使用csv.reader()函数和for循环来逐行处理数据;在写入CSV文件时,我们可以使用csv.writer()函数和csv_writer.writerows()函数来将数据写入CSV文件中。
希望本文对你在Python中使用csv模块读取和写入CSV文件有所帮助。祝学习愉快!
以上就是Python 3.x 中如何使用csv模块读取和写入CSV文件的详细内容,更多请关注每日运维网(www.mryunwei.com)其它相关文章!