web-dev-qa-db-ja.com

PythonによるYAMLファイルの編集

次のようなYAMLファイルがあります。

# Sense 1
- name  : sense1
  type  : float
  value : 31

# sense 2
- name  : sense2
  type  : uint32_t
  value : 1488

# Sense 3
- name  : sense3
  type  : int32_t
  value : 0

- name  : sense4
  type  : int32_t
  value : 0
- name  : sense5
  type  : int32_t
  value : 0
- name  : sense6
  type  : int32_t
  value : 0

Pythonを使用してこのファイルを開き、いくつかの値を変更して（上記を参照）、ファイルを閉じたいのですが、どうすればよいですか？

たとえば、YAML出力を同じにして、sense2 [value] = 1234に設定したいとします。

pythonyamledit
12
Tzimkiyahoo Bar Kozyva
with open("my_file.yaml") as f:
     list_doc = yaml.load(f)

for sense in list_doc:
    if sense["name"] == "sense2":
         sense["value"] = 1234

with open("my_file.yaml", "w") as f:
    yaml.dump(list_doc, f)
10
jwilner

マッピングキーの順序を維持する必要がある場合は、コメントとルートレベルのシーケンスの要素間の空白（例：このファイルはリビジョン管理下にあるため、 ruamel.yaml を使用する必要があります（免責事項：私はそのパッケージの作成者です）。

YAMLドキュメントがファイルinput.yamlにあると仮定します。

import sys
import ruamel.yaml

yaml = ruamel.yaml.YAML()
# yaml.preserve_quotes = True
with open('input.yaml') as fp:
    data = yaml.load(fp)
for elem in data:
    if elem['name'] == 'sense2':
         elem['value'] = 1234
         break  # no need to iterate further
yaml.dump(data, sys.stdout)

与える：

# Sense 1
- name: sense1
  type: float
  value: 31

# sense 2
- name: sense2
  type: uint32_t
  value: 1234

# Sense 3
- name: sense3
  type: int32_t
  value: 0

- name: sense4
  type: int32_t
  value: 0
- name: sense5
  type: int32_t
  value: 0
- name: sense6
  type: int32_t
  value: 0
2
Anthon