技术文摘
Python 写入 CSV 时 writerow() 和 writerows() 函数示例讲解
Python 写入 CSV 时 writerow() 和 writerows() 函数示例讲解
在 Python 中,处理 CSV(逗号分隔值)文件是常见的操作,而writerow()和writerows()函数则是csv模块中用于写入数据的重要方法。
让我们来了解一下writerow()函数。它用于将一行数据写入 CSV 文件。以下是一个简单的示例:
import csv
data = ['John', 'Doe', 25]
with open('example.csv', 'w', newline='') as file:
writer = csv.writer(file)
writer.writerow(data)
在上述代码中,我们首先创建了一个包含数据的列表data,然后打开一个名为example.csv的文件,并指定写入模式w和newline=''来处理换行符。通过创建csv.writer对象,调用writerow()方法将data列表中的数据写入文件。
接下来是writerows()函数,它用于将多行数据一次性写入 CSV 文件。示例如下:
import csv
data = [['Alice', 'Smith', 28], ['Bob', 'Johnson', 30]]
with open('example.csv', 'w', newline='') as file:
writer = csv.writer(file)
writer.writerows(data)
在这个例子中,data是一个包含多个列表的列表,每个内部列表代表一行数据。使用writerows()函数可以方便地将这些行数据同时写入 CSV 文件。
需要注意的是,在使用这两个函数时,要确保数据的格式符合 CSV 的规范,通常是每个数据项之间用逗号分隔。
另外,还可以通过设置delimiter参数来更改分隔符,例如,如果希望使用制表符作为分隔符,可以这样写:
writer = csv.writer(file, delimiter='\t')
writerow()函数适合逐行写入数据,而writerows()函数适用于一次性写入多行数据。根据实际需求选择合适的函数,可以更高效地完成 CSV 文件的写入操作。无论是处理少量数据还是大量数据,这两个函数都能为我们提供便利。
希望通过以上的示例讲解,您能清晰地理解和掌握writerow()和writerows()函数在 Python 中写入 CSV 文件的用法。
TAGS: Python写入CSV Python函数 CSV操作 函数示例