1. 程式人生 > >1.3預處理與熱圖

1.3預處理與熱圖

1 # 匯入後續所需要的庫 2 from sklearn.datasets import load_iris 3 from sklearn.preprocessing import scale 4 import numpy as np 5 import matplotlib.pyplot as plt 6 7 # 載入資料集 8 data = load_iris() 9 x = data['data'] 10 y = data['target'] 11 col_names = data['feature_names'] 12 13 # 資料預處理 14
# 根據平均值對資料進行縮放 15 x = scale(x, with_std=False) 16 x_ = x[1:26,] # 選取其中25組資料 17 y_labels = range(1, 26) 18 19 # 繪製熱圖 20 plt.close('all') 21 plt.figure(1) 22 fig, ax = plt.subplots() 23 ax.pcolor(x_, cmap=plt.cm.Greens, edgecolors='k') 24 ax.set_xticks(np.arange(0, x_.shape[1])+0.5) # 設定橫縱座標 25
ax.set_yticks(np.arange(0, x_.shape[0])+0.5) 26 ax.xaxis.tick_top() # x軸提示顯示在圖形上方 27 ax.yaxis.tick_left() # y軸提示顯示在圖形的左側 28 ax.set_xticklabels(col_names, minor=False, fontsize=10) # 傳遞標籤資料 29 ax.set_yticklabels(y_labels, minor=False, fontsize=10) 30 plt.show()