1. 程式人生 > >python資料視覺化--散點圖以及分類

python資料視覺化--散點圖以及分類

在繪製《機器學習實戰》第六章的簡單二分類資料時,想用散點圖來繪製,但是始終沒有結果,還好,花了半天的時間搞清楚了大致流程。特作筆記如下。

先看效果圖:

注意點一:

在書中P95頁,得到dataArr,和labelArr兩個資料,首先要對它們進行預處理:拿到的資料是普通矩陣形式的,而scatter函式的引數是array型別,所以要進行格式轉換:

dataArr_c=numpy.array(dataArr)

labelArr_c=numpy.array(labelArr)

注意點二:

拿到的labelArr_c是1,-1的值,需要將其轉為1,2的標籤值:

labelArr_c[labelArr_c == -1] = 2

否則,呼叫scatter後會得到如下結果:

全部程式碼如下:

from numpy import *  
import matplotlib  
import matplotlib.pyplot as plt

dataArr_c=numpy.array(dataArr)
labelArr_c=numpy.array(labelArr)
labelArr_c[labelArr_c == -1] = 2

f1 = plt.figure(1)  
plt.subplot(211)  
plt.scatter(dataArr_c[:,0],dataArr_c[:,1])
plt.subplot(212)   
label = labelArr_c
plt.scatter(dataArr_c[:,0],dataArr_c[:,1],15.0*labelArr_c,15.0*labelArr_c)