在 Python 中漂亮地将 2D 数组转储到 YAML

Question

假设我有一个像这样定义的二维数组：

import sys

from ruamel.yaml import YAML

yaml = YAML()
yaml.version = (1,2)

def main():
    data = {
        "foo": [
            [1,2,3],
            [4,5,6]
        ]
    }
    yaml.dump(data, sys.stdout)

if __name__ == '__main__':
    main()

我想输出一个“可读”、有效的 YAML 文件，其中每个“行”位于单独的行上：

"foo":
- [1,2,3]
- [4,5,6]

甚至

"foo": [
  [1,2,3],
  [4,5,5]
]

我已经研究过ruamel.yaml，但默认行为是每一列都在单独的行上，虽然有效，但不易阅读：

%YAML 1.2
---
foo:
- - 1
  - 2
  - 3
- - 4
  - 5
  - 6

Answer 1

当您将属性设置.default_flow_style为None（而不是默认值False）时，您的叶节点将以流样式表示：

import sys
import ruamel.yaml


yaml = ruamel.yaml.YAML()
yaml.version = (1,2)
yaml.default_flow_style = None

def main():
    data = {
        "foo": [
            [1,2,3],
            [4,5,6]
        ]
    }
    yaml.dump(data, sys.stdout)

if __name__ == '__main__':
    main()

这使：

%YAML 1.2
---
foo:
- [1, 2, 3]
- [4, 5, 6]

但上面的内容适用于整个文件。

如果您希望将单个列表表示为 YAML 中的流式序列，则应该将它们设为 类型，然后可以设置每个对象的CommentedSeq属性。这也是“知道”如何在往返时保留序列样式的方式：ruamel.yaml

import sys
import ruamel.yaml

yaml = ruamel.yaml.YAML()
yaml.version = (1,2)

a = ruamel.yaml.comments.CommentedSeq([1,2,3])
a.fa.set_flow_style()


def main():
    data = {
        "foo": [
        a,
            [4,5,6]
        ]
    }
    yaml.dump(data, sys.stdout)

if __name__ == '__main__':
    main()

给予：

%YAML 1.2
---
foo:
- [1, 2, 3]
- - 4
  - 5
  - 6