使用 python 将 Json 转换为 CSV 文件

我觉得你的方法还行.您只是在for key in keys循环中犯了一个错误:您当前所做的是 for each 元素创建一个DICT({key: current[key]}).所以在最后你会有一个out列表，这是一个不相关的词典的列表，其中每个videoID在一个词典中，type在一个不同的词典中. 就像这样:

[{'videoID': 'vid001'}, {'type': 'video'}, {'videoID': 'vid002'}, {'type': 'video'}, {'videoID': 'vid003'}, {'type': 'video'}, {'videoID': 'vid006'}, {'type': 'video'}, {'videoID': 'vid007'}, {'type': 'video'}, {'videoID': 'vid004'}, {'type': 'video'}, {'videoID': 'vid005'}, {'type': 'video'}, {'videoID': 'vid008'}, {'type': 'video'}, {'videoID': 'vid009'}, {'type': 'video'}, {'videoID': 'vid010'}, {'type': 'video'}]

相反，你想要的是:

[{'videoID': 'vid001', 'type': 'video'}, {'videoID': 'vid002', 'type': 'video'}, {'videoID': 'vid003', 'type': 'video'}, {'videoID': 'vid006', 'type': 'video'}, {'videoID': 'vid007', 'type': 'video'}, {'videoID': 'vid004', 'type': 'video'}, {'videoID': 'vid005', 'type': 'video'}, {'videoID': 'vid008', 'type': 'video'}, {'videoID': 'vid009', 'type': 'video'}, {'videoID': 'vid010', 'type': 'video'}]

其中每个VIDEO ID与其类型相关.

要做到这一点，你只需要在循环关键字时创建一个字典，将每个元素添加到字典中，然后在关键字循环结束时将该字典附加到out列表中.

我会做的是:

data_couple = {}
for key in keys:
    if key in current:
        data_couple[key] = current[key]
        # out.append({key: current[key]})
out.append(data_couple)``

因此，整个extract个函数将变成:

def extract(data, keys):
    out = []
    queue = [data]
    while len(queue) > 0:
        current = queue.pop(0)
        if type(current) == dict:
            data_couple = {}
            for key in keys:
                if key in current:
                    data_couple[key] = current[key]
            out.append(data_couple)
            for val in current.values():
                if type(val) in [list, dict]:
                    queue.append(val)
        elif type(current) == list:
            queue.extend(current)
    return out

最后，要在CSV文件中编写Dict，我只需使用CSV DictWriter:

import csv

def writeToCsv(dictionary, col_name):
    with open("file.csv", "w") as f:
        wr = csv.DictWriter(f, fieldnames=col_name)
        wr.writeheader()
        for elem in dictionary:
            wr.writerow(elem)
    
writeToCsv(your_dictionary, ["videoID", "type"])

这将创建一个具有videoID和type列的CSV 或者，正如Ivan Calderon Answer here所建议的，你也可以在一行中使用Pandas to_csv的方法:

在此之前，您应该更改您的词典:

x = extract(data, ["videoID","type"])
d = {k: [v] for k, v in x.items()}
pd.DataFrame.from_dict(data=d, orient='columns').to_csv('dict_file.csv')

但我对此不是很确定.