python sklearn 如何用测试集数据画出决策树（非开发样本）

Posted 2023-05-15

参考技术A #coding=utf-8

from sklearn.datasets import load_iris
from sklearn import tree

iris = load_iris()
clf = tree.DecisionTreeClassifier()
clf = clf.fit(iris.data, iris.target)

from sklearn.externals.six import StringIO
import pydot

dot_data = StringIO()
tree.export_graphviz(clf, out_file=dot_data)
graph = pydot.graph_from_dot_data(dot_data.getvalue())
graph[0].write_dot('iris_simple.dot')
graph[0].write_png('iris_simple.png')

神级程序员教你如何用四行代码画出一幅中国地图！Python就是牛逼

前面两行引入相应的库，真正的代码就4行，够简单吧。第1行甚至可以不写，它定义了图的大小。第2行我们创建一个地图，第3行把海岸线画上，第4行显示这个地图，就是这样：

你用 Java 的 4 行代码画一个地图出来？

然后我们开始画上国家，又是1行代码：

m.drawcountries(linewidth=1.5)

就变成了这样：

看上去有点变形，这是因为我们没有添加任何投影的原因， Basemap 提供 24 种不同的投影方式，你可以自己一个个试一下，比较常用的是 兰勃特投影 ，我们添加一下：

m = Basemap(llcrnrlon=77, llcrnrlat=14, urcrnrlon=140, urcrnrlat=51, projection=‘lcc‘, lat_1=33, lat_2=45, lon_0=100)

这次终于看上去比较正常了：