如何在 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，例如：

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


Out[]:
"Message 5"

如何在 Python 中创建和遍历决策树

How to create and traverse decision tree in Python

python

python-3.x

tree

decision-tree

diagram