決定木を使用して散布図の境界線を設定する

Question

csvのデータを用いて，散布図（x,y）を作成します。
値に応じて境界線を引きたく、決定木を使用して作成したいのですがうまくいきません。
色分けとしては、値の小さい順から
'white', 'green', 'purple', 'sienna', 'deepskyblue', 'orange', 'red', 'darkgrey'

【CSVデータ】１列目：x、2列目：y、３列目：値
https://d.kuku.lu/e1245ec59c

xの幅は190-300、yの幅は0-15です。

図の作成方法を教えてください。よろしくお願いします。

import os
import sys
import numpy
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import matplotlib.cm as cm
from matplotlib.colors import ListedColormap
import glob
from PIL import Image
import PIL
from sklearn import tree
from sklearn.tree import DecisionTreeClassifier

path = "/media/student/data1"
path1 = "/media/student/data1/csv/"

datalist1 = glob.glob(path1 + "*.csv")

for data1 in datalist1:
    name1 = data1[len(path1)+4:-4]

    if name1 :
        print name1 + "..." + "OK"
	    cloud = pd.read_csv(data1)
	    clf = DecisionTreeClassifier(max_depth=2, random_state = 0) # make instance max_depth:tree depth
	    my_cmap = ListedColormap(['white', 'green', 'purple', 'sienna', 'deepskyblue', 'orange', 'red', 'darkgrey'])
	    plt.title("title")
	    plt.xlabel("X")
	    plt.ylabel("Y")
	    plt.imshow(x=cloud[:,1], y=cloud[:,2], c=cloud[:,3], cmap=my_cmap , extent=(190,300,0,12), interpolation='none')
	    pil_img = Image.fromarray(np.uint8(clf)) 
        print "cloud...ok"
        outdata = path + "/outdata5/"  +name + ".png"
        plt.savefig(outdata)

途中まで作成したコードです。間違っている部分が多いので
参考程度でよろしくお願い申し上げます。

Answer 1

こんな感じでしょうか。決定境界はちょっとむずいので、分類結果の領域ごとに背景色を変えています。

あなたの環境やデータで直接動作するコードではないので、参考程度でよろしくおねがいします。

# このへんも参考に
# https://scikit-learn.org/stable/auto_examples/classification/plot_classifier_comparison.html#sphx-glr-auto-examples-classification-plot-classifier-comparison-py


import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from matplotlib.colors import ListedColormap
from sklearn.tree import DecisionTreeClassifier

# アップされているデータをdata.csvとして保存しました。
cloud = pd.read_csv("data.csv", header=None)

dt = DecisionTreeClassifier(max_depth=None, random_state=0)
dt.fit(cloud.iloc[:,0:2], cloud[2])

x = np.arange(190, 300, 0.1)
y = np.arange(0, 10, 0.01)
xx, yy = np.meshgrid(x, y)

zz = dt.predict(np.vstack([xx.ravel(), yy.ravel()]).T).reshape(xx.shape)

my_cmap = ListedColormap(['white', 'green', 'purple', 'sienna', 'deepskyblue', 'orange', 'red', 'darkgrey'])
plt.title("title")
plt.xlabel("X")
plt.ylabel("Y")
plt.scatter(x=cloud[0], y=cloud[1], c=cloud[2], cmap=my_cmap)
plt.contourf(xx, yy, zz, cmap=my_cmap, alpha=.2)
plt.savefig("result.png")

Comments

[py-python]

ありがとうございます。出来ました！