1.matplitlib绘制热图
1 import random 2 from matplotlib import pyplot as plt 3 from matplotlib import cm 4 from matplotlib import axes 5 6 def draw(): 7 # define the x and y axis of the hotmap 8 xLabel = ['A', 'B', 'C', 'D', 'E'] 9 yLabel = ['1', '2', '3', '4', '5'] 10 11 # prepaer the data, generate the two-dimension data 12 data = [] 13 for i in range(5): 14 temp = [] 15 for j in range(5): 16 k = random.randint(0,100) 17 temp.append(k) 18 data.append(temp) 19 20 # plot the figure 21 fig = plt.figure() 22 ax = fig.add_subplot(111) 23 # define the scale 24 ax.set_yticks(range(len(yLabel))) 25 ax.set_yticklabels(yLabel, fontproperties=plt.cm.hot_r) 26 ax.set_xticks(range(len(xLabel))) 27 ax.set_xtickslabels(xLabel) 28 # make the figure and select the style of hotmap 29 im = ax.imshow(data, cmap = plt.cm.hot_r) # the value is more high, the color is more deep 30 # im = ax.imshow(data, cmap = plt.cm.hot) # the value is more high, the color is more shallow 31 # plt.cm.~ hot, cool, gray, bone, white, spring, summer, autumn, winter 32 33 # add the scale bar of the right site 34 plt.colorbar(im) 35 36 plt.title("This is a title") 37 plt.show() 38 39 d = draw()
2.Seaborn绘制热图
import numpy as np import seaborn as sns import matplotlib.pyplot as plt sns.set() # data 1 np.random.seed(0) uniform_data = np.random.rand(10, 12, vmin = 0, vmax = 1, center = 0) # vmin=0, vmax=1 : the scope of colorbar value # center=0 : colorbar valuee centered at 0 ax = sns.heatmap(uniform_data) # data 2 flights_long = sns.load_dataset("flights") flights = flights_long.pivot("month", "year", "passengers") f, ax = plt.subplots(figsize = (9, 6)) # annot = True: show the value sns.heatmap(flights, annot = True, fmt = 'd', cmap = 'Y|GnBu', linewidth = 5, ax = ax) label_y = ax.get_yticklabels() plt.setp(label_y, rotation = 360, horizontalalignment = 'right') label_x = ax.get_xticklabels() plt.setp(label_x, rotation = 45, horizontalalignment = 'right') plt.show() # https://www.jianshu.com/p/d105d934d876