Bagaimana saya bisa menulis data dalam format YAML dalam sebuah file?

130

Saya perlu menulis data di bawah ini ke file yaml menggunakan Python:

{A:a, B:{C:c, D:d, E:e}}

yaitu kamus dalam kamus. Bagaimana saya bisa mencapai ini?

— pengguna1643521
sumber

Lihat juga: Bagaimana saya bisa mengurai file YAML dengan Python

— Martin Thoma

Jawaban:

196

import yaml

data = dict(
    A = 'a',
    B = dict(
        C = 'c',
        D = 'd',
        E = 'e',
    )
)

with open('data.yml', 'w') as outfile:
    yaml.dump(data, outfile, default_flow_style=False)

The default_flow_style=Falseparameter yang diperlukan untuk menghasilkan format yang Anda inginkan (gaya aliran), jika tidak untuk koleksi bersarang menghasilkan blok gaya:

A: a
B: {C: c, D: d, E: e}

— Matthew Trevor
sumber

default_flow_style = True melakukan yang sebaliknya seperti yang dinyatakan dalam jawaban di bawah ini!

— encc

Apakah ada manfaat menggunakan di withsini dibandingkan dengan satu baris?

— src

@src withmemastikan file ditutup dengan benar ketika tidak lagi diperlukan

— kbolino

Lihat Munch , stackoverflow.com/questions/52570869/… import yaml; from munch import munchify; f = munchify(yaml.safe_load(…));print(f.B.C)

— Hans Ginzel

Tautan ke dokumentasi PyYAML yang menunjukkan perbedaan default_flow_styleparameter. Untuk menulisnya ke file dalam mode blok (seringkali lebih mudah dibaca):

d = {'A':'a', 'B':{'C':'c', 'D':'d', 'E':'e'}}
with open('result.yml', 'w') as yaml_file:
    yaml.dump(d, yaml_file, default_flow_style=False)

menghasilkan:

A: a
B:
  C: c
  D: d
  E: e

— lou
sumber

Lihat Munch , stackoverflow.com/questions/52570869/… import yaml; from munch import munchify; f = munchify(yaml.safe_load(…));print(f.B.C)

— Hans Ginzel