创建一个智能函数来从字典列表中提取值

荣耀9211

我有一个这样的数据列表

data = [
          {
                    "name": "Box 0",
                    "type": "Box",
                    "vals": {
                              "corner1": "0,0,0",
                              "corner2": "0,0,5",
                              "rotate": "0",
                              "scale": "1",
                              "translate": "0,0,0"
                    }
          },
          {
                    "name": "Ovus 1",
                    "type": "Ovus",
                    "vals": {
                              "radiusb": "1",
                              "radiust": "0.5",
                              "rotate": "0",
                              "scale": "1",
                              "translate": "0,0,0"
                    }
          }
]

我想创建一个表单函数

fetch_attributes(name, key)

我可以在其中给出关键属性值类型、值或比例。

我的问题是 scale 属性在 vals 字典中。我可以让函数在我给key=type时给我类型，当我要求key=scale 时，它会自动进入 vals 字典。

我的方法

我使用此代码作为 name = 'Box 0' 的示例

obj = [i for i in data if i['name'] == name][0]
key_1 = "['vals']" # If I want complete dict of vals
key_2 = "['vals']['scale']" #If I want only one val out of vals
key_3 = "['type']" # If I want type

result_1 = eval('obj' + key_1)
result_2 = eval('obj' + key_2)
result_3 = eval('obj' + key_3)

我想知道是否存在执行此任务的更好或更 Pythonic 的方法

编辑

我想要一种方法，它也将有助于或指导创建一个 edit_attributes 函数作为

edit_attributes(name, key, value)

马蒂诺

我认为这（现在）做你想要的：

def fetch_attributes(data, name, key):
    for dct in data:
        if dct['name'] == name:
            obj = dct
            break
    else:
        raise KeyError(f'No object with name {name!r} in data')

    if isinstance(obj, dict):
        if key in obj:
            return obj[key]

        for value in obj.values():
            if isinstance(value, dict):  # Nested dict?
                if key in value:
                    return value[key]

    raise KeyError(f'{key!r} not in data')


print(fetch_attributes(data, 'Box 0', 'vals'))   # -> {'corner1': '0,0,0', 'corner2': '0,0,5', 'rotate': '0', 'scale': '1', 'translate': '0,0,0'}
print(fetch_attributes(data, 'Box 0', 'scale'))  # -> 1
print(fetch_attributes(data, 'Box 0', 'type'))   # -> Box

本文收集自互联网，转载请注明来源。

如有侵权，请联系 [email protected] 删除。