matplotlib——熱圖hot map contour() 以及圖像存儲

＃！/ usr / bin / env python
 ＃ -  *  -  coding：utf-8  -  *  -
 import numpy as np
 import matplotlib.pyplot as plt


def main（）：
    ＃熱圖＃
     plt.figure（ 1）  ＃建立一個數字＃
     plt.subplot（ 331）  ＃3行和3列＃的圖
    數據= np.random.rand（ 3， 3）
    plt.imshow（data，interpolation = 'nearest'，cmap = plt.get_cmap（'cool'），aspect = 'auto'，vmin = 0，vmax = 1）
     ＃imshow（）函數實現熱圖繪製
     ＃＃X存儲圖像，可以是浮點型數組，unit8數組以及PIL圖像，如果其爲數組，則需滿足一下形狀：
     ＃＃（1）M * N此時數組必須爲浮點型，其中值爲該座標的灰度;
     ＃（2）M * N * 3 RGB（浮點型或者單位8類型）＃＃
     （3）M * N * 4 RGBA（浮點型或者單位8類型）
     ＃＃想要某塊顯示成一種顏色，則需要調用interpolation ='nearest'參數
     ＃參數cmap用於設置熱圖的Colormap。（參考百度百科）Colormap是MATLAB裏面用來設定和獲取當前色圖的函數＃
    ＃默認情況下，imshow將數據標準化爲最小和最大值。＃使用vmin和vmax參數或norm參數來控制（如果您想要非線性縮放）
     ＃＃hot map＃
     plt.subplot（ 313）

    def（x，y）：
         return（ 1 -x / 2 + x ** 5 + y ** 3）* np.exp（-x ** 2 -y ** 2）
    N = 256
     X = np.linspace（ - 3，3，n）的
    Y = np.linspace（ - 3，3，n）的
    X，Y = np.meshgrid（x，y）
    plt.contour（X，Y，f（X，Y），8，alpha = .75，camp = plt.get_cmap（'hot'））
    plt.savefig（“E：\ untitled2 / data / fig1.png”）
    plt.show（）
if__name__ == “ __main__ ”：
    main（）