错误信息显示的是行号(我假设它是9)。指向那条线
if elem['name'] == 'Stuart':
而且,如果这没有给您任何提示,那么我在这种情况下建议的方法就是开始添加一些
for循环是这样的:
for elem in data: print('elem', elem) if elem['name'] == 'Stuart': print('elem->hobbies', elem['hobbies']) elem['hobbies'] = ['Fishing']
此打印
elem team_member
在引发异常之前,我希望这使您意识到自己不是在遍历列表的 元素 (项目),而是遍历dict的 键 (由YAML的根级别映射构成)。与键相关联的 值
是具有键
name和键的对象
hobbies。
因此,请更改变量
elem以
key明确您要处理的内容,然后继续使用
value与该键关联的值,而不是
elem在该循环中¹:
for key in data: value = data[key] if value['name'] == 'Stuart': print('value->hobbies', value['hobbies']) value['hobbies'] = ['Fishing']
这给出:
value->hobbies ['dancing']team_member: name: Max hobbies: - Readingteam_leader: name: Stuart hobbies: - Fishing
因此,我们摆脱了异常,但是结果却不完全是您想要的。
dancing密钥“爱好”的元素消失了,因为您为该密钥分配了一个新的(列表)值,而您应该做的是将一个项目附加到列表中。现在我们也可以摆脱打印功能:
for key in data: value = data[key] if value['name'] == 'Stuart': value['hobbies'].append('Fishing')
这将使您在文件的最后顺序中得到两项。还有其他一些事情要解决:
- 大写
dancing
不正确。若要更正此问题,如果只有一个元素,则添加一行处理列表 - 名称的代码
Max
需要添加(这就是为什么需要删除break
代码中的的原因) - 空行被视为对第一个序列的最后一个元素的注释,该注释需要移动
- 您的序列缩进不是默认值
最终代码如下:
from pathlib import Pathimport ruamel.yamlpath = Path('input.yaml')yaml = ruamel.yaml.YAML()yaml.indent(sequence=4, offset=2) # for the non-default indentation of sequencesdata = yaml.load(path)for key in data: value = data[key] if value['name'] == 'Stuart': if len(value['hobbies']) == 1: value['hobbies'][0] = value['hobbies'][0].capitalize() value['hobbies'].append('Fishing') elif value['name'] == 'Max': last_item_index = len(value['hobbies']) - 1 value['hobbies'].append('Painting') comments = value['hobbies'].ca if not comments.items[last_item_index][0].value.strip(): # move empty comment lines from previous last item to new last item comments.items[last_item_index + 1] = comments.items.pop(last_item_index)yaml.dump(data, path)
这使您获得的东西非常接近您想要的
team_member: name: Max hobbies: - Reading - Paintingteam_leader: name: Stuart hobbies: - Dancing - Fishing
¹前两行的替代方法:
for key, value in data.items()