Как я могу напечатать [] без строки в Python в файле YAML

Question

В моем файле ям я пытаюсь это

with open(fname, "w") as f:
     yaml.safe_dump({'items':['test', 'test2']}, f,
                    default_flow_style=False, width=50, indent=4)

Он печатает в следующем формате

items:
- 'test'
- 'test2'

Я хочу, чтобы вывод был отформатирован, как показано ниже

items: ['test', 'test2']

Как я могу это сделать?

РЕДАКТИРОВАТЬ:

Это мой полный код

   d = {}        
   for m in ['B1', 'B2', 'B3']:
                d2 = {}
                for f in ['A1', 'A2', 'A3']:
                    # here i don't want any flow style
                    d2[f] = ['test', 'test2']
                d[m] = d2

    with open(fname, "w") as f:
        yaml.safe_dump(d, f, default_flow_style=True, width=50, indent=8)

Answer 1

Не кладите default_flow_style=False затем делает полную противоположность того, что вы хотите:

>>> import yaml
>>> yaml.safe_dump({'items': ['test', 'test2']}, default_flow_style=False)
'items:\n- test\n- test2\n'
>>> yaml.safe_dump({'items': ['test', 'test2']})
'items: [test, test2]\n'

Что касается частичного форматирования документа, вы можете сделать это с помощью пользовательских представлений, например:

class Items(list):
    pass


def items_representer(dumper, data):
    return dumper.represent_sequence('tag:yaml.org,2002:seq', data, flow_style=True)


yaml.representer.SafeRepresenter.add_representer(Items, items_representer)

result = yaml.safe_dump({
    'items': Items(['test', 'test2']),
    'other list': ['1', '2'],
}, default_flow_style=False)

# items: [test, test2]
# other list:
# - '1'
# - '2'
print(result)

Answer 2

Если вы хотите точный контроль и имеете только определенные списки со стилем потока, вы должны использовать ruamel.yaml (это моя улучшенная версия PyYAML):

import ruamel.yaml
from ruamel.yaml.comments import CommentedSeq

d = {}
for m in ['B1', 'B2', 'B3']:
    d2 = {}
    for f in ['A1', 'A2', 'A3']:
        # here i don't want any flow style
        d2[f] = CommentedSeq(['test', 'test2'])
        if f != 'A2':
            d2[f].fa.set_flow_style()
    d[m] = d2

x = ruamel.yaml.dump(
    d, Dumper=ruamel.yaml.RoundTripDumper,
    default_flow_style=False, width=50, indent=8)
print(x)

дам тебе:

B1:
        A1: [test, test2]
        A3: [test, test2]
        A2:
        - test
        - test2
B2:
        A1: [test, test2]
        A3: [test, test2]
        A2:
        - test
        - test2
B3:
        A1: [test, test2]
        A3: [test, test2]
        A2:
        - test
        - test2

CommentedSeq ведет себя так же, как обычный список Python, но позволяет вам указать комментарий, установить стиль потока / блока и т. д.

ruamel.yaml обычно используется для сохранения комментариев, стиля потока / блока для элементов и т. д. при циклическом отключении YAML. Т.е. если бы вы добавили:

d2 = ruamel.yaml.load(x, Loader=ruamel.yaml.RoundTripLoader)
y = ruamel.yaml.dump(
    d2, Dumper=ruamel.yaml.RoundTripDumper, width=50, indent=8)
assert x == y

утверждение верно.

Но, конечно, он также может быть использован для создания YAML с нуля. Например, вы могли бы также использовать тип CommentedMap и сохранять порядок ваших слов.