我正在尝试在python中创建一个树,如下图所示:
如何用Python编写代码来创建此树并找到给定的“答案”或叶节点(给定性别和颜色)。
例如:
Input: { Gender : "Female", Color : "Green" }
Output: "Message 5"
我将添加更多的级别和节点,所以我尝试创建树表示而不是一堆“ if”语句,因为那样可能会很杂乱。
答案 0 :(得分:2)
您可以按dict
结构创建树,例如:
tree = {'Male': {'Red': 'Message 1', 'Green': 'Message 2', 'Blue': 'Message 3'},
'Female': {'Red': 'Message 4', 'Green': 'Message 5', 'Blue': 'Message 6'}}
然后遍历此树仅是关键查找,例如:
In []:
i = {'Gender' : "Female", 'Color' : "Green" }
tree[i['Gender']][i['Color']]
Out[]:
"Message 5"