Python中如何創建和遍歷決策樹

Question

我正在嘗試在 python 中創建一個如下圖所示的樹：

我如何在 Python 中編寫代碼來創建這棵樹並在給定性別和顏色的情況下找到給定的“答案”或葉節點。

例如：

Input: { Gender : "Female", Color : "Green" }

Output: "Message 5"

我將添加更多級別和節點，因此我嘗試創建一個樹表示而不是一堆“if”語句，因為那樣可能會很混亂。

Answer 1

您可以在dict結構中創建樹，例如：

tree = {'Male': {'Red': 'Message 1', 'Green': 'Message 2', 'Blue': 'Message 3'}, 
        'Female': {'Red': 'Message 4', 'Green': 'Message 5', 'Blue': 'Message 6'}}

那么遍歷這棵樹無非就是查key，eg：

In []:
i = {'Gender' : "Female", 'Color' : "Green" }
tree[i['Gender']][i['Color']]


Out[]:
"Message 5"