Question

我想为我的Python脚本的某些分析设置提供YAML配置文件。 YAML文件看起来像这样：

settings:
  param1: 1
  param2: two
  ...

在我的Python脚本中，我想将这些值作为关键字参数传递给函数。在YAML文件中加载会给我一本字典。该函数只有关键字参数，我不知道配置文件将具有哪些关键字参数。该函数看起来像这样：

def process(param1=None, param2=None, param3=None):
    ...

请注意，YAML设置的名称与关键字参数名称相同。该函数有几个可能的关键字参数，配置文件可能只有一个或两个。显然，我想将值分配给正确的关键字参数。

如何将YAML值（和/或字典）分配给关键字参数分配？

Answer 1

您可以从yaml.loads()获取字典，并使用**语法将键/值传递给函数。这样您就不必事先知道密钥。例如：

import yaml

with open('path/to/config.yml', 'rb') as f:
    conf = yaml.loads(f.read())    # load the config file

def process(**params):    # pass in variable numbers of 
    for key, value in params.items():
        print('%s: %s' % (key, value))

process(**conf)    # pass in your keyword args

尝试使用您的代码，它应该足以让您了解如何实现自己的代码。

Answer 2

你可以在python中迭代dict中的键：

for key in data:
    print(key)

这将允许您通过调用：

获取密钥并在函数中使用它

data[key] # returns value from dict

Answer 3

所以，我跑过你的问题，我认为这可能会有所帮助。 PyYAML太棒了！它支持的一件事是直接从python

进行类映射

import yaml
class Hero:
    def __init__(self, name, hp, sp):
        self.name = name
        self.hp = hp
        self.sp = sp
    def __repr__(self):
        return "Hero %s has %d Hitpoints" % (self.name, self.hp)


f = "sample.yml"
with open(f, mode='w') as ff:
    ff.write(yaml.dump(Hero(name="Cool Guy", hp=4, sp=10)))

    # saves as !!python/object:__main__.Hero {hp: 4, name: Cool Guy, sp: 10}

    ff.close()
with open(f, mode="r") as fx:
    y = yaml.load(fx)
    print(y)

[out]: Hero Cool Guy has 4 Hitpoints

希望你能像我一样找到尽可能多的用途。有关详细信息，请查看PyYAML Docs。

将YAML配置值输入到Python的关键字参数中

3 个答案: