在python中将yaml解析为列表

问题描述 投票:0回答:2

我需要在项目中使用 YAML。我有一个 YAML 文件,我用它来填充与数据交互的 Python 程序中的列表。

我的数据如下所示:

Employees:
    custid: 200
    user: Ash
        - Smith
        - Cox

我需要 python 代码来迭代此 YAML 文件并填充如下列表:

list_of_Employees = ['Ash', 'Smith' 'Cox']

我知道我必须打开文件,然后将数据存储在变量中,但我不知道如何在列表中单独输入每个元素。理想情况下,我想使用附加功能,这样我就不必每次都确定我的“用户”大小。

python list yaml pyyaml
2个回答
12
投票
with open("employees.yaml", 'r') as stream:
    out = yaml.load(stream)
    print out['Employees']['user']

这应该已经为您提供了用户列表。另请注意,您的 YAML 在用户节点后缺少一个破折号。


9
投票

YAML 序列会为您转换为 Python 列表(至少在使用 PyYAMLruamel.yaml 时),因此您无需自己添加任何内容。

在 PyYAML 和 ruamel.yaml 中,您要么将文件/流传递给

load()
例程,要么将字符串传递给它。两者:

import ruamel.yaml

with open('input.yaml') as fp:
    data = ruamel.yaml.load(fp)

import ruamel.yaml

with open('input.yaml') as fp:
    str_data = fp.read()
data = ruamel.yaml.load(str_data)

做同样的事情。

假设您将输入更正为:

Employees:
    custid: 200
    user:
    - Ash
    - Smith
    - Cox

可以打印出数据:

{'Employees': {'custid': 200, 'user': ['Ash', 'Smith', 'Cox']}}

并看到列表已经存在并且可以通过正常的字典查找访问:

data['Employees']['user']

© www.soinside.com 2019 - 2024. All rights reserved.