欢迎访问宙启技术站
智能推送

Python处理CSV与List的转换方法

发布时间:2023-05-17 07:35:43

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数据进行操作。