Python把csv数据写入list和字典类型的变量脚本方法

在Python中，我们可以使用csv模块来读取和写入CSV文件。CSV文件是一种常见的数据格式，通常用于存储表格数据。本文将介绍如何将CSV数据写入list和字典的变量。

将CSV数据写入list

我们可以使用csv模块的writer()函数将CSV数据写入list中。下面是一个示例：

import csv

# 写入CSV数据到list
data = [['Tom', 20, 'M'], ['Jerry', 21, 'M'], ['Mickey', 22, 'F']]
with open('data.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(data)

# 读取CSV数据到list
with open('data.csv', 'r') as file:
    reader = csv.reader(file)
    data = list(reader)

# 输出数据
print(data)

在这个示例中，我们使用csv.writer()函数将list数据写入CSV文件。然后我们使用csv.reader()函数将数据读取到list中，并输出结果。

将CSV数据写入字典

我们也可以使用csv模块的DictWriter()函数CSV数据写入字典中。下面是一个示例：

import csv

# 写入CSV数据到字典
data = [{'name': 'Tom', 'age': 20, 'gender': 'M'},
        {'name': 'Jerry', 'age': 21, 'gender': 'M'},
        {'name': 'Mickey', 'age': 22, 'gender': 'F'}]
with open('data.csv', 'w', newline='') as file:
    fieldnames = ['name', 'age', 'gender']
    writer = csv.DictWriter(file, fieldnames=fieldnames)
    writer.writeheader()
    writer.writerows(data)

# 读取CSV数据到字典
with open('data.csv', 'r') as file:
    reader = csv.DictReader(file)
    data = [row for row in reader]

# 输出数据
print(data)

在这个示例中，我们使用csv.DictWriter()函数将字典数据写入CSV文件。然后我们csv.DictReader()函数将数据读取到字典中，并输出结果。

总结

本文介绍了如何将CSV数据写入list和字典的变量。我们可以使用csv模块的writer()函数和DictWriter()函数将list和字典数据写入CSV文件。在实际应用中，需要根据具体的需求选择适合的方法，以便更好地处理数据。