Python3中绘制决策树的模块

Question

我想问一下如何用python绘制决策树。我的意思是功能类似于this question，但是需要图片，而不仅仅是文本格式。你知道吗

例如，拥有这个字典结构

MyNodeStructure = {'main': {'load': {'regA': None},
                            'comupute': {'loop': {'regA': None,
                                                  'regB': None}},
                            'store': None}}

我想得到这样的输出： Example of desired tree。你知道吗

你知道如何在Python3中实现这个结果的模块/简单方法吗？你知道吗