Python处理CSV与List的转换方法
Python是一种很强大的编程语言,在数据处理方面有着广泛的应用。CSV(Comma Separated Values)是一种常见的文件格式,用于存储表格数据。在实际开发中,经常会遇到需要将CSV文件转换为List(列表)的情况。本文将介绍Python处理CSV与List的转换方法。
1. CSV文件的读取与写入
读取CSV文件可以使用Python内置的csv模块。csv模块提供了Reader和Writer两个对象,分别用于读取和写入CSV文件。
读取CSV文件的方法如下所示:
import csv
with open('file.csv', 'r') as f:
reader = csv.reader(f)
for row in reader:
print(row)
在上述代码中,使用open()方法打开CSV文件,并将其作为参数传递给csv.reader()函数。然后使用for循环逐行读取文件中的数据。
如果要将数据写入CSV文件,可以使用csv.writer()函数。示例如下:
import csv
data = [['Name', 'Age', 'Gender'], ['Tom', 20, 'Male'], ['Lily', 18, 'Female']]
with open('file.csv', 'w', newline='') as f:
writer = csv.writer(f)
writer.writerows(data)
在上述代码中,使用open()方法创建CSV文件,并将其作为参数传递给csv.writer()函数。然后使用writerows()方法将数据写入文件中。
2. CSV转换为List
将CSV文件转换为List,可以先使用上述方法读取CSV文件,然后将每行数据转换为List。示例如下:
import csv
data = []
with open('file.csv', 'r') as f:
reader = csv.reader(f)
for row in reader:
data.append(row)
print(data)
在上述代码中,首先创建了一个空列表data。然后使用for循环逐行读取CSV文件中的数据,并将其追加到data列表中。最后打印data列表即可。
3. List转换为CSV
将List转换为CSV文件,可以先创建一个csv.writer()对象,然后使用writerows()方法写入数据。示例如下:
import csv
data = [['Name', 'Age', 'Gender'], ['Tom', 20, 'Male'], ['Lily', 18, 'Female']]
with open('file.csv', 'w', newline='') as f:
writer = csv.writer(f)
writer.writerows(data)
在上述代码中,首先创建了一个包含表格数据的二维列表data。然后使用open()方法创建CSV文件,并将其作为参数传递给csv.writer()函数创建一个写入器使用。最后通过writerows()方法将数据写入文件中。
总结
Python处理CSV与List的转换方法非常简单,只需要使用csv模块提供的读取和写入函数即可。读取CSV文件时,使用csv.reader()函数逐行读取文件中的数据,并将其转换为List;写入CSV文件时,使用csv.writer()函数将List转换为CSV表格数据,并写入CSV文件中。通过这些方法,我们可以很方便地处理CSV文件中的数据,并将其转换为List数据进行操作。
