python树状结构可视化

• plotTree.py

# -*- coding: utf-8 -*-

import sys  
import os
import numpy as np
import matplotlib.pyplot as plt
import treePlotter as tp 



# 绘制树

myTree = {
   'root': {
   0: 'leaf node', 1: {
   'level 2': {
   0: 'leaf node', 1: 'leaf node'}},2:{
   'level2': {
   0: 'leaf node', 1: 'leaf node'}}}}
tp.createPlot(myTree)

• treePlotter.py

'''
Created on Oct 14, 2010

@author: Peter Harrington
'''
import matplotlib.pyplot as plt

decisionNode = dict(boxstyle="sawtooth", fc="0.8")
leafNode = dict(boxstyle="round4", fc="0.8")
arrow_args = dict(arrowstyle="<-")

def getNumLeafs(myTree):
    numLeafs = 0
    firstStr = list(myTree.keys())[0]
    secondDict = myTree[firstStr]
    for key in secondDict.keys():
        if