如何在Python的yaml文件中配置多个字典键?

时间:2018-08-10 15:39:30

标签: python yaml pyyaml

我正在尝试使用yaml文件存储字典键,以便我可以解析各种json负载。

我将以下yaml文件加载到名为“ cfg”的变量中:

my_table: metadata

当我跑步时:

for i in json['resources']:
    print(i[cfg['my_table']])

我可以从json的['metadata']键中看到所有值。

但是,我想从['resources']下的 ['metadata'] ['guid'] 键中查看所有值。因此,当我将yaml文件更改为:

my_table: metadata, guid

它不起作用。我还尝试了这些变体:

my_table: ['metadata', 'guid']  #tried ''.join("['{0}']".format(a) for a in cfg['my_table']) which gives me what I want, but when I shove this inside i[], I get a key error
my_table: ['metadata']['guid']  #yaml file doesn't parse properly

有什么想法吗?

1 个答案:

答案 0 :(得分:1)

那么您想将字典关键字存储在yaml文件中吗?

就做

# keys.yaml
my_table: [metadata, guid]

然后在您的python文件中

keys = yaml.safe_load('keys.yaml')['my_table']

for obj in json['resources']:  # don't call it json, that's a package name
    print(obj[keys[0]][keys[1]])

对于可变数量的键(不仅是两个),请按照以下步骤操作:

def recursive_getitem(obj, keys):
   if len(keys) == 0:
      return obj
   return recursive_getitem(obj[keys[0]], keys[1:])

然后做

for obj in json['resources']:  
    print(recursive_getitem(obj, keys))

要在评论中回答第二个问题,您需要使用嵌套列表

# keys.yaml
my_table:
    - [metadata, guid]
    - [entity, name]

# py file
keys = yaml.safe_load('keys.yaml')['my_table']

for obj in json['resources']:  # don't call it json, that's a package name
    print(obj[keys[0][0]][keys[0][1]])
    print(obj[keys[1][0]][keys[1][1]])