python - 字典循环获取每个值

Question

没有任何导入

# given
deps = {'W': ['R', 'S'], 'C': [], 'S': ['C'], 'R': ['C'], 'F': ['W']}
prob = {'C': [0.5], 'R': [0.2, 0.8], 'S': [0.5, 0.1], 'W': [0.01, 0.9, 0.9,     0.99], 'F' : [0.4, 0.3]}
k = 'F'
# want to return:  L = [[0.2, 0.8], [0.5, 0.1], [0.01, 0.9, 0.9, 0.99], [0.4, 0.3]]


# attempt

L = []

for i in deps[k]:
    s = i
    while(deps[s] != []):
        L.append(prob[s])
        s = deps[s]
print(L)

我很难弄明白这一点。因此，给定 2 个字典:依赖项和概率，我希望遍历一个选择点并设置每个值，因此对于上面的示例，我选择了“F”。

它会首先进入 'F' 的部门，找到 'W' 然后检查它的部门 ['R', 'S'] 然后检查 'R' 看到 'R' 的 depedent 是'C' 和 'C' 不是依赖项，所以我们在 'R' 处停止并将其概率附加到 L。

 [[0.2, 0.8]]

然后我们进入S并做同样的事情

[[0.2, 0.8], [0.5, 0.1]]

然后我们就完成了，我们回到了W

[[0.2, 0.8], [0.5, 0.1], [0.01, 0.9, 0.9, 0.99]]

最后因为我们完成了 W 我们得到了 F 的概率

[[0.2, 0.8], [0.5, 0.1], [0.01, 0.9, 0.9, 0.99], [0.4, 0.3]]

当存在多个相关值时，我的代码会失败。不确定如何解决这个问题。尝试创建一个函数，在给定 deps 和 prob 以及 k 值的情况下执行此操作

Answer 1

我会用一个 while 循环来解决这个问题，这个循环不断地查看您是否已经使用了您递归找到的所有值。您可以使用如下结构:

deps = {'W': ['R', 'S'], 'C': [], 'S': ['C'], 'R': ['C'], 'F': ['W']}
# out = ['F', 'W', 'R', 'S']
prob = {'C': [0.5], 'R': [0.2, 0.8], 'S': [0.5, 0.1], 'W': [0.01, 0.9, 0.9, 0.99], 'F': [0.4, 0.3]}
k = 'F'

L = []
my_list = []
found_all = False

def get_values(dep_dictionary, prob_dict, start_key):
    used_keys = []
    keys_to_use = [start_key]
    probability = []
    # build a list of linked values from deps dictionary
    while used_keys != keys_to_use:
        print('used: {}'.format(used_keys))
        print('to use: {}'.format(keys_to_use))
        for i in range(len(keys_to_use)):
            if keys_to_use[i] not in used_keys:
                new_keys = dep_dictionary[keys_to_use[i]]
                if len(new_keys):
                    for sub_key in new_keys:
                        if sub_key not in keys_to_use:
                            keys_to_use.append(sub_key)
                    used_keys.append(keys_to_use[i])
                else:
                    del keys_to_use[i]
    # at this point used_keys = ['F', 'W', 'R', 'S']
    for key in used_keys:
        probability.append(prob_dict[key])
    print(probability)

get_values(deps, prob, k)

哪些输出:

used: []
to use: ['F']
used: ['F']
to use: ['F', 'W']
used: ['F', 'W']
to use: ['F', 'W', 'R', 'S']
used: ['F', 'W', 'R', 'S']
to use: ['F', 'W', 'R', 'S', 'C']
[[0.4, 0.3], [0.01, 0.9, 0.9, 0.99], [0.2, 0.8], [0.5, 0.1]]

哪里可以看到输出是正确的([[0.4, 0.3], [0.01, 0.9, 0.9, 0.99], [0.2, 0.8], [0.5, 0.1]]),然而它的顺序并不完全相同，但听起来这不应该是一个大问题。如果是，您始终可以通过调整

for key in used_keys:
    probability.append(prob_dict[key])

使得 probability 也是一个字典。您还可以取出 print() 语句，它们只是用来调试和直观地显示循环中发生的事情。你也可能有函数 return probability 而不是打印它，但我会留给你自行决定!