Question

这是我的YAML文件（input.yaml）：

team_member:
  name: Max
  hobbies:
    - Reading

team_leader:
  name: Stuart
  hobbies:
    - dancing

我想编辑此YAML文件以在关键的“爱好”中添加更多值，例如：

team_member:
  name: Max
  hobbies:
    - Reading
    - Painting

team_leader:
  name: Stuart
  hobbies:
    - Dancing
    - Fishing

我尝试实现代码Anthon以适合我的情况，但是它根本没有帮助，因为该YAML文件的缩进级别与我的不同。
示例：

import sys
import ruamel.yaml

yaml = ruamel.yaml.YAML()
# yaml.preserve_quotes = True
with open('input.yaml') as fp:
    data = yaml.load(fp)
for elem in data:
    if elem['name'] == 'Stuart':
         elem['hobbies'] = ['Fishing']
         break  # no need to iterate further
yaml.dump(data, sys.stdout)

我收到错误消息“ TypeError（'字符串索引必须为整数'，）”，我知道这段代码可能是完全错误的，但是我对ruamel.yaml不熟悉。

该如何编码？

Answer 1

错误信息显示的是行号（我假设它是9）。指向该行

    if elem['name'] == 'Stuart':

如果这没有给您任何线索，那么在这种情况下，我推荐的方法是开始添加一些print函数，以便您知道自己在做什么。 for循环如下：

for elem in data:
    print('elem', elem)
    if elem['name'] == 'Stuart':
         print('elem->hobbies', elem['hobbies'])
         elem['hobbies'] = ['Fishing']

此打印

 elem team_member

在引发异常之前，我希望这会使您意识到您不是在遍历列表的 elem 项（项目），而是在 key 上进行迭代的s（从YAML中的根级别映射构造）。与键关联的 value 是具有键name和键hobbies的对象。

因此，将变量elem更改为key，以弄清您要处理的内容，然后继续使用value（与该键关联的值而不是{{1}） }在该循环中¹：

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: - Reading team_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 Path
import ruamel.yaml

path = Path('input.yaml')
yaml = ruamel.yaml.YAML()
yaml.indent(sequence=4, offset=2)  # for the non-default indentation of sequences

data = 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
- Painting

team_leader:
name: Stuart
hobbies:
- Dancing
- Fishing}

Answer 2

感谢Anthon您的代码有效，我必须按如下方式编辑此代码：

$xml = <<<'XML'
<subject id="Tom" xmlns="http://cpee.org/ns/organisation/1.0">
  <relation unit="ITSupport" role="ITSupporter" />
</subject>
XML;

$role = 'ITSupporter';

$document = new DOMDocument();
$document->loadXML($xml);
$xpath = new DOMXpath($document);
// register a prefix for the namespace
$xpath->registerNamespace('org', 'http://cpee.org/ns/organisation/1.0');

$ids = [];
// address the elements using the registered prefix
$idAttributes = $xpath->evaluate("//org:subject[org:relation/@role='".$role."']/@id");
foreach ($idAttributes as $idAttribute) {
  $ids[] = $idAttribute->value;
}
var_dump($ids);

如何读取YAML文件中的组件，以便可以使用ruamel.yaml编辑它的键值？

2 个答案: