博客

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

  • Post category:Python

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

将CSV数据写入list

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

import csv

# 打开CSV文件
with open('data.csv', 'r') as file:
    # 创建CSV读取器
    reader = csv.reader(file)
    # 将数据读取到list中
    data = list(reader)

# 输出数据
print(data)

在这个示例中，我们打开了一个名为data.csv的CSV文件，并使用csv.reader()函数创建了一个CSV读取器。然后我们将数据读取到list中，并输出结果。

将CSV数据写字典

我们也可以使用csv模块的DictReader()函数将CSV数据读取到字典中。下面是一个示例：

import csv

# 打开CSV文件
with open('data.csv', 'r') as file:
    # 创建CSV读取器
    reader = csv.DictReader(file)
    # 将数据读取到字典中
    data = [row for row in reader]

# 输出数据
print(data)

在这个示例中，我们打开了一个名为data.csv的CSV文件，并使用csv.DictReader()函数创建了一个CSV读取器。然后我们将数据读取到字典中，并输出结果。

将list或字典数据写入CSV文件

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

import 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)

# 写入字典数据
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.writer()函数和csv.DictWriter()函数将list和字典数据写入CSV文件。注意，在使用csv.writer()函数时，我们需要使用writerows()函数将整个list写入文件。而在使用csv.DictWriter()函数时，我们需要使用writeheader()函数写入表头，并使用writerows()函数将整个字典列表写入文件。

总结

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

Tags: