如何使用 python 向 yaml 文件添加新行/更新数据？

Question

我想更新 my.yaml 文件，并在每次迭代中将新数据添加到 my.yaml 文件中，同时仍保存以前的数据，这是我的一段代码：

import yaml

num=0
for i in range(4):
    num +=1    
    data_yaml =[{"name" : num,  "point" : [x, y , z]}]

    with open('points.yaml', 'w') as yaml_file:
        yaml.dump(data_yaml, yaml_file)  

这是我想要在我的points.yaml文件中实现的目标 output 结果：

- name: 1
  point: [0.7, -0.2, 0.22]
- name: 2
  point: [0.6, -0.11, 0.8]
- name: 3
  point: [0.4, -0.2, 0.6]
- name: 4
  point: [0.3, -0.7, 0.8]
- name: 5
  point: [0.1, -0.4, 0.2]

如何 append 或在.yaml 文件中的先前数据旁边自动添加新行？

Answer 1

在预期的 output 中，您的根级数据结构是一个序列。 因此，在您的 Python 程序中，您应该从一个空列表开始。 （如果您不知道，最简单的方法是加载您手工制作的.load文档，看看它是如何最终成为数据结构 Python 的。）

您似乎不仅使用的是 EOL 的 Python 版本，而且还使用了 ruamel.yaml 的旧（兼容性）例程。 如果不能改变前者，至少开始使用新的 ruamel.yaml API：

from __future__ import print_function

import sys
import ruamel.yaml

points = [
  [0.7, -0.2, 0.22],
  [0.6, -0.11, 0.8],
  [0.4, -0.2, 0.6],
  [0.3, -0.7, 0.8],
  [0.1, -0.4, 0.2],
]

data = []

yaml = ruamel.yaml.YAML(typ='safe')

num=0
for i in range(5):
    num +=1
    x, y, z = points[i]
    data.append({"name" : num,
    "point" : [x, y , z ]
    })

    with open('points.yaml', 'w') as yaml_file:
         yaml.dump(data, yaml_file)

with open('points.yaml') as yaml_file:
    print(yaml_file.read())

这使：

- name: 1
  point: [0.7, -0.2, 0.22]
- name: 2
  point: [0.6, -0.11, 0.8]
- name: 3
  point: [0.4, -0.2, 0.6]
- name: 4
  point: [0.3, -0.7, 0.8]
- name: 5
  point: [0.1, -0.4, 0.2]

请注意，我将range()的参数更改为 5。