如何在 Python 中将 CSV 文件转换为 JSON 文件

Python 1年前 1.1K

CSV（Comma-Separated Values）文件是一种常见的数据存储格式，它将数据以逗号分隔的形式存储在纯文本文件中。而 JSON（JavaScript Object Notation）文件则是一种轻量级的数据交换格式，它以键值对的形式存储数据，并且易于解析和生成。在数据处理和数据分析中，将 CSV 文件转换为 JSON 文件是一个常见的操作。本文将介绍如何使用 Python 将 CSV 文件转换为 JSON 文件，并且提供一些注意事项。

导入模块

在 Python 中，我们可以使用 csv 模块和 json 模块来处理 CSV 文件和 JSON 文件。在代码开始之前，我们需要先导入这两个模块。

import csv
import json

读取 CSV 文件

在将 CSV 文件转换为 JSON 文件之前，我们需要先读取 CSV 文件。我们可以使用 csv 模块中的 reader() 函数来读取 CSV 文件。该函数返回一个迭代器对象，我们可以通过循环遍历迭代器来获取 CSV 文件中的每一行数据。

with open('data.csv', 'r') as file:
    reader = csv.reader(file)
    for row in reader:
        print(row)

在上面的代码中，我们打开名为 data.csv 的 CSV 文件，并使用 csv.reader() 函数读取该文件。然后，我们使用 for 循环遍历迭代器对象，获取 CSV 文件中的每一行数据，并打印出来。

转换为 JSON 格式

在读取 CSV 文件之后，我们需要将其转换为 JSON 格式。我们可以使用 json 模块中的 dump() 函数将 CSV 数据转换为 JSON 格式，并将其写入到一个新的 JSON 文件中。

data = []
with open('data.csv', 'r') as file:
    reader = csv.DictReader(file)
    for row in reader:
        data.append(row)

with open('data.json', 'w') as file:
    json.dump(data, file)

在上面的代码中，我们首先创建一个空列表 data，用于存储 CSV 文件中的数据。然后，我们打开名为 data.csv 的 CSV 文件，并使用 csv.DictReader() 函数读取该文件。该函数返回一个迭代器对象，每一行数据都被转换为一个 OrderedDict 对象。我们通过循环遍历迭代器对象，将每一行数据添加到 data 列表中。

接下来，我们打开一个新的文件 data.json，并使用 json.dump() 函数将 data 列表中的数据转换为 JSON 格式，并写入到该文件中。

示例

下面是一个完整的示例代码，用于将名为 data.csv 的 CSV 文件转换为名为 data.json 的 JSON 文件。

import csv
import json

data = []
with open('data.csv', 'r') as file:
    reader = csv.DictReader(file)
    for row in reader:
        data.append(row)

with open('data.json', 'w') as file:
    json.dump(data, file)