将长度可变的嵌套数组输出到CSV

import csv

def write_nested_array_to_csv(nested_array, csv_file):
    with open(csv_file, 'w', newline='') as file:
        writer = csv.writer(file)
        
        # 写入CSV头部
        headers = get_headers(nested_array)
        writer.writerow(headers)
        
        # 写入数据
        write_data(nested_array, writer)

def get_headers(nested_array):
    headers = set()
    for item in nested_array:
        if isinstance(item, dict):
            headers.update(item.keys())
    return list(headers)

def write_data(nested_array, writer):
    for item in nested_array:
        if isinstance(item, dict):
            row = [item.get(header, '') for header in writer.fieldnames]
            writer.writerow(row)
        elif isinstance(item, list):
            write_data(item, writer)

# 示例数据
nested_array = [
    {'name': 'Alice', 'age': 25, 'city': 'New York'},
    {'name': 'Bob', 'age': 30, 'city': 'San Francisco'},
    [
        {'name': 'Charlie', 'age': 35, 'city': 'Seattle'},
        {'name': 'Dave', 'age': 40, 'city': 'Chicago'}
    ]
]

# 输出到CSV文件
write_nested_array_to_csv(nested_array, 'output.csv')