python树状结构取值和加值

# 循环拿key为objkey的value值
# data代表目标数据，objkey代表对应的key，store_data存储value，一般传[]
def get_value(data, objkey, store_data):
    if isinstance(data, dict):
        for k, v in data.items():
            if k == objkey and v:
                store_data.append(v)
            else:
                if isinstance(v, dict) or isinstance(v, list):
                    get_value(v, objkey, store_data)
    elif isinstance(data, list):
        for i in data:
            if isinstance(data, dict) or isinstance(data, list):
                get_value(i, objkey, store_data)
    return store_data


# 为dict树状结构添加key：value
def add_tree_value(data, key, add_data):
    for data_i in data:
        if key in data_i:
            data_i.update(add_data)
            if data_i[key]:
                add_tree_value(data_i[key], key, add_data)
    return data