Python--常见的数据格式转换

下面是几个常见的数据格式转换的示例，涵盖了一些常用的格式，如 CSV、XML、YAML 等。每个示例都会介绍如何从一种格式转换到另一种格式。

1. CSV 转 JSON

CSV 文件通常以逗号分隔，行代表记录，列代表字段。我们可以使用 csv 和 json 模块来实现转换。

示例代码

import csv
import json

# 定义输入和输出文件路径
csv_file = 'data.csv'
json_file = 'output.json'

def csv_to_json(csv_file, json_file):
    data = []
    
    # 读取 CSV 文件并转换为字典列表
    with open(csv_file, mode='r', encoding='utf-8') as infile:
        reader = csv.DictReader(infile)
        for row in reader:
            data.append(row)
    
    # 将字典列表写入 JSON 文件
    with open(json_file, 'w', encoding='utf-8') as outfile:
        json.dump(data, outfile, indent=4, ensure_ascii=False)
    
    print(f"CSV to JSON conversion complete! Output saved to {json_file}")

# 调用函数
csv_to_json(csv_file, json_file)

示例输入（CSV 文件 data.csv）

name,age,city
John,30,New York
Jane,25,Los Angeles

输出（JSON 文件 output.json）

[
    {
        "name": "John",
        "age": "30",
        "city": "New York"
    },
    {
        "name": "Jane",
        "age": "25",
        "city": "Los Angeles"
    }
]

---

2. JSON 转 CSV

从 JSON 文件转换为 CSV 文件，可以使用 Python 的 csv 模块写出 CSV 格式。

示例代码

import json
import csv

# 定义输入和输出文件路径
json_file = 'data.json'
csv_file = 'output.csv'

def json_to_csv(json_file, csv_file):
    with open(json_file, 'r', encoding='utf-8') as infile:
        data = json.load(infile)
    
    # 提取字段名（假设所有 JSON 对象都有相同的字段）
    fieldnames = data[0].keys()

    # 将数据写入 CSV 文件
    with open(csv_file, 'w', newline='', encoding='utf-8') as outfile:
        writer = csv.DictWriter(outfile, fieldnames=fieldnames)
        writer.writeheader()
        writer.writerows(data)
    
    print(f"JSON to CSV conversion complete! Output saved to {csv_file}")

# 调用函数
json_to_csv(json_file, csv_file)

示例输入（JSON 文件 data.json）

[
    {"name": "John", "age": 30, "city": "New York"},
    {"name": "Jane", "age": 25, "city": "Los Angeles"}
]

输出（CSV 文件 output.csv）

name,age,city
John,30,New York
Jane,25,Los Angeles

---

3. XML 转 JSON

XML 是一种常用于数据交换的格式，Python 中可以使用 xmltodict 模块来处理 XML 数据并转换为 JSON。

示例代码

import xmltodict
import json

# 定义输入和输出文件路径
xml_file = 'data.xml'
json_file = 'output.json'

def xml_to_json(xml_file, json_file):
    # 读取 XML 文件
    with open(xml_file, 'r', encoding='utf-8') as infile:
        xml_content = infile.read()
        data_dict = xmltodict.parse(xml_content)  # 将 XML 转换为字典

    # 将字典写入 JSON 文件
    with open(json_file, 'w', encoding='utf-8') as outfile:
        json.dump(data_dict, outfile, indent=4, ensure_ascii=False)
    
    print(f"XML to JSON conversion complete! Output saved to {json_file}")

# 调用函数
xml_to_json(xml_file, json_file)

示例输入（XML 文件 data.xml）

<people>
    <person>
        <name>John</name>
        <age>30</age>
        <city>New York</city>
    </person>
    <person>
        <name>Jane</name>
        <age>25</age>
        <city>Los Angeles</city>
    </person>
</people>

输出（JSON 文件 output.json）

{
    "people": {
        "person": [
            {
                "name": "John",
                "age": "30",
                "city": "New York"
            },
            {
                "name": "Jane",
                "age": "25",
                "city": "Los Angeles"
            }
        ]
    }
}

---

4. YAML 转 JSON

YAML 是一种简洁的配置文件格式。可以使用 PyYAML 库将 YAML 文件转换为 JSON 文件。

示例代码

import yaml
import json

# 定义输入和输出文件路径
yaml_file = 'data.yaml'
json_file = 'output.json'

def yaml_to_json(yaml_file, json_file):
    # 读取 YAML 文件
    with open(yaml_file, 'r', encoding='utf-8') as infile:
        data = yaml.safe_load(infile)  # 将 YAML 加载为 Python 字典

    # 将字典写入 JSON 文件
    with open(json_file, 'w', encoding='utf-8') as outfile:
        json.dump(data, outfile, indent=4, ensure_ascii=False)
    
    print(f"YAML to JSON conversion complete! Output saved to {json_file}")

# 调用函数
yaml_to_json(yaml_file, json_file)

示例输入（YAML 文件 data.yaml）

people:
  - name: John
    age: 30
    city: New York
  - name: Jane
    age: 25
    city: Los Angeles

输出（JSON 文件 output.json）

{
    "people": [
        {
            "name": "John",
            "age": 30,
            "city": "New York"
        },
        {
            "name": "Jane",
            "age": 25,
            "city": "Los Angeles"
        }
    ]
}

---

5. JSON 转 YAML

可以使用 PyYAML 库将 JSON 文件转换为 YAML 文件。

示例代码

import json
import yaml

# 定义输入和输出文件路径
json_file = 'data.json'
yaml_file = 'output.yaml'

def json_to_yaml(json_file, yaml_file):
    # 读取 JSON 文件
    with open(json_file, 'r', encoding='utf-8') as infile:
        data = json.load(infile)

    # 将数据写入 YAML 文件
    with open(yaml_file, 'w', encoding='utf-8') as outfile:
        yaml.dump(data, outfile, allow_unicode=True)
    
    print(f"JSON to YAML conversion complete! Output saved to {yaml_file}")

# 调用函数
json_to_yaml(json_file, yaml_file)

示例输入（JSON 文件 data.json）

{
    "people": [
        {
            "name": "John",
            "age": 30,
            "city": "New York"
        },
        {
            "name": "Jane",
            "age": 25,
            "city": "Los Angeles"
        }
    ]
}

输出（YAML 文件 output.yaml）

people:
- age: 30
  city: New York
  name: John
- age: 25
  city: Los Angeles
  name: Jane

---

总结

这些示例展示了如何在不同常见数据格式之间进行转换，包括 CSV、JSON、XML 和 YAML。这些操作在数据处理、配置文件转换、数据导入导出等任务中非常常用。