在字典中的字典中抓取值

时间:2015-07-10 16:13:38

标签: python dictionary

我可能没有编写最好的代码,但是假设我在字典中有一个字典,其输出如下:

print node
> {
>  0: {'NODE': 'CLOUDY', 'GATE': 'EQUAL', 'PROB': 0.5}, 
>  1: {'NODE': 'RAIN', 'CPT_1': 0.2, 'CPT_0': 0.8, 'GATE': 'OR', 'PROB': 0.5, 'PNODE_0': 'Cloudy'}, 
>  2: {'NODE': 'SPRINKLER', 'CPT_1': 0.9, 'CPT_0': 0.5, 'GATE': 'OR', 'PROB': 0.7, 'PNODE_0': 'Cloudy'}, 
>  3: {'NODE': 'WETGRASS', 'CPT_3': 0.01, 'CPT_2': 0.1, 'CPT_1': 0.1, 'CPT_0': 1.0, 'GATE': 'OR', 'PROB': 0.5, 'PNODE_0': 'Rain', 'PNODE_1': 'Sprinkler'}
> }

我希望看到最终结果:

print string
> ['CLOUDY', 'RAIN', 'SPRINKLER', 'WETGRASS']

到目前为止,我已经写了这个以获得与上面输出完全相同的输出。

string = "["
for value in xrange (0, len(node)):
    string += "'" + node[value].get("NODE") + "', "
string = string[:-2]
string += "]"
print string

我正在绕着一个字典中的字典提取所有值,这个字典中包含一个“NODE”键。这可能吗?

1 个答案:

答案 0 :(得分:2)

只需使用dict中的值:

print([d["NODE"] for d in node.values()])
['CLOUDY', 'RAIN', 'SPRINKLER', 'WETGRASS']

如果密钥可能不在所有字样中,只需先使用in进行检查:

print([d["NODE"] for d in node.values() if "NODE" in d])

node.values()会给你所有的答案:

[{'NODE': 'CLOUDY', 'GATE': 'EQUAL', 'PROB': 0.5}, {'NODE': 'RAIN', 'CPT_1': 0.2, 'CPT_0': 0.8, 'GATE': 'OR', 'PROB': 0.5, 'PNODE_0': 'Cloudy'}, {'NODE': 'SPRINKLER', 'CPT_1': 0.9, 'CPT_0': 0.5, 'GATE': 'OR', 'PROB': 0.7, 'PNODE_0': 'Cloudy'}, {'NODE': 'WETGRASS', 'CPT_3': 0.01, 'CPT_1': 0.1, 'CPT_0': 1.0, 'CPT_2': 0.1, 'GATE': 'OR', 'PROB': 0.5, 'PNODE_0': 'Rain', 'PNODE_1': 'Sprinkler'}]

所以你只需从每个中提取。

如果您确实想要形成一个看起来像列表的字符串:

 s = "[{}]".format(",".join([d["NODE"] for d in node.values() if "NODE" in d]))

或简单:

print(str([d["NODE"] for d in node.values() if "NODE" in d]))