在 Python 中将 yml 解析为 csv 格式

Question

我有一个大约百万行的 yml 文件。 它具有以下结构：

    categories:
    - id: "927"
      depth: "1"
      active: "Y"
      name:
        ru: Строительство и ремонт
      is_merged: "1"
      sort: "21"
      properties:
      - id: "77"
        filter: 1
        top: 1
        filter_sort: "500"
        display_type: F
      - id: "79"
        filter: 1
        filter_sort: "500"
        display_type: F
      - id: "8013"
        display_type: F
      - id: "13694"
        filter: 1
        filter_sort: "1"
        expanded: 1
        display_type: F
        values:
        - key: "95574"
          xmlid: ep
          rank: "500"
          value:
            ru: Эп
        - key: "95576"
          xmlid: other
          rank: "1000"
          value:
            ru: Другой

请帮助我将其转换为 csv 格式并将其解析为每个步骤中包含列的文件，例如：
categories | id | depth | active | name | ru | is_merged | sort | properties | id | filter | top | filter_sort | display_type | values | key | xmlid | rank | value | ru

Answer 1

好吧，这很难。

这是我到目前为止所拥有的。

首先，为了摆脱 yaml 文件中的列表，我手动删除了破折号（“-”）。 因为列表很烦人，无法在文件中获取“键值”对。

不以手动执行此操作为荣。 如果有人可以编写一个函数来做到这一点，我会很乐意将它添加到我的答案中。

删除 yaml 文件中的破折号后，您可以尝试以下代码：

import csv
import yaml

key_list = []
value_list = []
final_value_list = []

# To get all the items in a nested dictionary
def recursive_items(dictionary):
    for key, value in dictionary.items():
        if type(value) is dict:
            yield (key, value)
            key_list.append(key)
            yield from recursive_items(value)
        else:
            yield (key, value)
            key_list.append(key)

# Load the Data, assuming the file name as "file.yml"
for filename in ['file.yml']:
    with open(filename) as f_input:
        data = yaml.safe_load(f_input)

# Get the pairs
for key, value in recursive_items(data):
    print(key, value)
    value_list.append(value)

# Adjust the value pairs in a new list.
for elem in value_list:
    if type(elem) == str or type(elem) == int:
        final_value_list.append(elem)
    else:
        final_value_list.append("")

# Write the results to a csv file
with open('output.csv', 'wt', newline ='') as data:
    writer = csv.writer(data, delimiter=',')
    writer.writerow(i for i in key_list)
    writer.writerow(j for j in final_value_list)

到目前为止的结果如下所示：