1. 程式人生 > >matplotlib繪製熱圖

matplotlib繪製熱圖

# -*- coding: utf-8 -*-

import numpy as np
import matplotlib.pyplot as plt

def plot_heat_map(x,figure_no):
    plt.figure(figure_no)
    plt.pcolor(x)
    plt.colorbar()

if __name__=='__main__':
    plt.close('all')
    x=np.random.normal(loc=0.5,scale=0.2,size=(10,10))

    plot_heat_map(x,2)