python笔记 --pydot安装与绘制决策树

1. 安装graphviz-2.28.0.msi
   下载地址：http://download.csdn.NET/detail/shouwangzhelv/9492517
   或者：csdn地址
   配置系统环境变量：C:\Program Files (x86)\Graphviz 2.28\bin添加到path中
2. 安装pyparsing-1.5.7.win32-py2.7.exe;
   遇到问题：Python version 2.7 required, which was not found
   解决办法在最后。
   https://pypi.python.org/pypi/pyparsing/1.5.7
3. pydot安装
   下载地址：https://pypi.python.org/pypi/pydot2/1.0.33然后安装，
   1）解压
   2）cmd 到包所在位置，注意必须是 setup.py所在路径
   3） 执行python setup.py install
4. 测试

import sys
reload(sys)
sys.setdefaultencoding("utf-8")
from sklearn.datasets import load_iris
from sklearn import tree
from sklearn.externals.six import StringIO
import pydot

iris = load_iris()#载入数据集
clf = tree.DecisionTreeClassifier()#算法模型
clf = clf.fit(iris.data, iris.target)#模型训练

with open("tree.dot", 'w') as f:
    f = tree.export_graphviz(clf, out_file=f)  
os.unlink('tree.dot') 
dot_data = StringIO()
tree.export_graphviz(clf, out_file=dot_data)

graph = pydot.graph_from_dot_data(dot_data.getvalue())
graph.write_pdf("iris.pdf")#写入pdf

执行完后显示ture
5. Python version 2.7 required, which was not found 解决方法

copy以下代码到register.py, 然后执行该文件， 出现：
python2.7 is now registered

#  
# script to register Python 2.0 or later for use with win32all  
# and other extensions that require Python registry settings  
#  
# written by Joakim Loew for Secret Labs AB / PythonWare  
#  
# source:  
# http://www.pythonware.com/products/works/articles/regpy20.htm  
#  
# modified by Valentine Gogichashvili as described in http://www.mail-archive.com/distutils-sig@python.org/msg10512.html  

import sys  

from _winreg import *  

# tweak as necessary  
version = sys.version[:3]  
installpath = sys.prefix  

regpath = "SOFTWARE\\Python\\Pythoncore\\%s\\" % (version)  
installkey = "InstallPath"  
pythonkey = "PythonPath"  
pythonpath = "%s;%s\\Lib\\;%s\\DLLs\\" % (  
    installpath, installpath, installpath  
)  

def RegisterPy():  
    try:  
        reg = OpenKey(HKEY_CURRENT_USER, regpath)  
    except EnvironmentError as e:  
        try:  
            reg = CreateKey(HKEY_CURRENT_USER, regpath)  
            SetValue(reg, installkey, REG_SZ, installpath)  
            SetValue(reg, pythonkey, REG_SZ, pythonpath)  
            CloseKey(reg)  
        except:  
            print "*** Unable to register!"  
            return  
        print "--- Python", version, "is now registered!"  
        return  
    if (QueryValue(reg, installkey) == installpath and  
        QueryValue(reg, pythonkey) == pythonpath):  
        CloseKey(reg)  
        print "=== Python", version, "is already registered!"  
        return  
    CloseKey(reg)  
    print "*** Unable to register!"  
    print "*** You probably have another Python installation!"  

if __name__ == "__main__":  
    RegisterPy()  

参考：
http://blog.sina.com.cn/s/blog_e15493ed0102wltx.html
http://blog.csdn.net/lulei1217/article/details/49583357