假設data是m行兩列的訓練樣本，labels是m行一列的類標簽，類標簽一共有3類，分別用1、2、3表示，現將data用散點圖表示出來，且不同類的樣本有不同的顏色：

import matplotlib.pyplot as pltfig = plt.figure() ax = fig.add_subplot(111) # 創建一個一行一列的圖 ax.scatter(data[:, 0], data[:, 1], 15.0*np.array(labels), 15.0*np.array(labels)) # 15.0是散點的大小 plt.show()

?

為了得到更好的效果，并以紅色的'*'表示類標簽1、藍色的'o'表示表示類標簽2、綠色的'+'表示類標簽3，修改參數如下：

import numpy as np import matplotlib.pyplot as pltfig = plt.figure() ax = fig.add_subplot(111) labels = np.array(labels) idx_1 = np.where(labels == 1) # 找出第一類 p1 = ax.scatter(data[idx_1, 0], data[idx_1, 1], marker='*', color='r', label='1',s=20) idx_2 = np.where(labels == 2) # 找出第二類 p2 = ax.scatter(data[idx_2, 0], data[idx_2, 1], marker = 'o',color ='b',label='2',s=10) idx_3 = np.where(labels == 3) # 找出第三類 p3 = ax.scatter(data[idx_3, 0], data[idx_3, 1], marker = '+',color ='g',label='3',s=30) plt.legend(loc='upper right') plt.show()

?

總結

以上是生活随笔為你收集整理的使用Matplotlib创建散点图的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。