python动态对象查询

时间:2016-10-04 04:11:38

标签: python arrays object dictionary

我有一个python对象,结构如下:

tree = {
    "a": {
        "aa": {
            "aaa": {},
            "aab": {},
            "aac": {}
        },
        "ab": {
            "aba": {},
            "abb": {}
        }
    },
    "b": {
        "ba": {}
    },
    "c": {}
}

一组字符串列表,如:

arr_1 = ["a", "ab"]
arr_2 = ["b", "ab"]
arr_3 = ["a", "ab", "aba"]

每个列表都定义了树中键的路径,我想确定列表是否描述了树中的有效路径。

目前,我可以像这样逐案回答:

tree[arr_1[0]][arr_1[1]] # Returns correct object
tree[arr_2[0]][arr_2[1]] # Returns error
tree[arr_3[0]][arr_3[1]][arr_3[2]] # Returns correct object

虽然这不符合我的要求。我更喜欢一个函数,它会在树中搜索任何给定列表中的键。

以下功能几乎是我想要的,但它不处理不同长度的列表。

def is_valid(tree, arr):
    obj = tree[arr[0]][arr[1]]
    if len(obj) > 0:
        return(obj)
    else:
        return("No obj")

目前此功能输出

is_valid(tree, arr_1) # Returns the correct result
is_valid(tree, arr_2) # Returns an error
is_valid(tree, arr_3) # Returns the same result as arr_1, which is incorrect

任何人都可以帮我扩展这个功能,以动态地对arr参数的长度作出反应吗?

谢谢!

1 个答案:

答案 0 :(得分:3)

我认为最简单的方法是使用递归。每个子树都是树,每个子路径都是路径,因此我们可以查看路径中的第一个元素是否有效,然后从那里继续。

def is_valid(tree, path):

    # base case. Any path would be valid for any tree
    if len(path) == 0:
        return True

    # the path must be invalid, no matter what comes before or after
    if not path[0] in tree:
        return False

    # look at the rest
    return is_valid(tree[path[0]], path[1:])

如果您希望子树描述路径,则可以执行此操作:

def is_valid(tree, path):

    if len(path) == 0:
        return tree # the only difference
    if not path[0] in tree:
        return False
    return is_valid(tree[path[0]], path[1:])