python中来自yaml的csv文件

Question

4:
- a: [0.6511928334730416, -0.6444996859128429, 0.40070930065859994]
  b: [-70.26386506067132, 19.563101216582368, 315.91419304402643]
  c: [462, 429, 401, 389]
  d: 1
5:
- e: [0.6511928334730416, -0.6444996859128429, 0.40070930065859994]
  f: [-70.26386506067132, 19.563101216582368, 315.91419304402643]
  g: [462, 429, 401, 389]
  h: 1

这是我的 yml 格式，我只需要从中提取键 c 和 g 的值并创建一个 csv 文件。 我是python的新手。
任何人都可以在这里指导我在此先感谢

Answer 1

通过循环你可以做到，但这只是一个你也可以拥有更好的想法

试试这个：-

import yaml
MyDict = yaml.load(open('example.yaml'))

    key = ['c','g']
    for i in MyDict.values():
        for j  in i:
            for k in key:
                if k in j:
                    print(j.get(k))

Answer 2

如果您的 yaml 转换为 dict 类似于：

d = {4:
       {
      'a': [0.6511928334730416, -0.6444996859128429, 0.40070930065859994],
      'b': [-70.26386506067132, 19.563101216582368, 315.91419304402643],
      'c': [462, 429, 401, 389],
      'd': 1
       },
    5:
     {'e': [0.6511928334730416, -0.6444996859128429, 0.40070930065859994],
      'f': [-70.26386506067132, 19.563101216582368, 315.91419304402643],
      'g': [462, 429, 401, 380],
      'h': 1
     }
    }

您可以尝试将以下代码转换为 CSV 值：

import csv  
import itertools  

# taking only the values of keys 'c' and 'g'
li = [i[j] for i in d.values()  for j in i.keys()  if j == "c" or j == "g"]

# the above list of list making a flat list for better result
merged = list(itertools.chain.from_iterable(li))

with open("filename.csv",'w') as myfile:
    wr = csv.writer(myfile, lineterminator='\n')
    # If you want first list to csv you can use li instead of merged 
    wr.writerow(merged)

Answer 3

我已经尝试并测试了这个

import yaml

with open(r'test.yaml') as file:
    documents = yaml.full_load(file)

for item, doc in documents.items():
    print(item, ":", doc)