【问题标题】:Parse yaml into a list in python将yaml解析为python中的列表
【发布时间】:2015-12-31 04:34:37
【问题描述】:
我需要为项目使用 YAML。我有一个 YAML 文件,我用它来填充与数据交互的 Python 程序中的列表。
我的数据如下所示:
Employees:
custid: 200
user: Ash
- Smith
- Cox
我需要 python 代码来遍历这个 YAML 文件并填充这样的列表:
list_of_Employees = ['Ash', 'Smith' 'Cox']
我知道我必须打开文件,然后将数据存储在一个变量中,但我不知道如何在列表中分别输入每个元素。理想情况下,我想使用 append 函数,这样我就不必每次都确定我的“用户”大小。
【问题讨论】:
标签:
python
list
yaml
pyyaml
【解决方案1】:
YAML 序列会为您翻译成 Python 列表(至少在使用 PyYAML 或 ruamel.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']}}
并看到该列表已经存在并且可以通过正常的 dict 查找访问:data['Employees']['user']
【解决方案2】:
with open("employees.yaml", 'r') as stream:
out = yaml.load(stream)
print out['Employees']['user']
应该已经为您提供了用户列表。还请注意,您的 yaml 在用户节点后缺少一个破折号