Python绘制热图

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 &＃61; [&＃39;A&＃39;, &＃39;B&＃39;, &＃39;C&＃39;, &＃39;D&＃39;, &＃39;E&＃39;]
9 yLabel &＃61; [&＃39;1&＃39;, &＃39;2&＃39;, &＃39;3&＃39;, &＃39;4&＃39;, &＃39;5&＃39;]
10
11 # prepaer the data, generate the two-dimension data
12 data &＃61; []
13 for i in range(5):
14 temp &＃61; []
15 for j in range(5):
16 k &＃61; random.randint(0,100)
17 temp.append(k)
18 data.append(temp)
19
20 # plot the figure
21 fig &＃61; plt.figure()
22 ax &＃61; fig.add_subplot(111)
23 # define the scale
24 ax.set_yticks(range(len(yLabel)))
25 ax.set_yticklabels(yLabel, fontproperties&＃61;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 &＃61; ax.imshow(data, cmap &＃61; plt.cm.hot_r) # the value is more high, the color is more deep
30 # im &＃61; ax.imshow(data, cmap &＃61; 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 &＃61; draw()

import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt
sns.set()
# data 1
np.random.seed(0)
uniform_data &＃61; np.random.rand(10, 12, vmin &＃61; 0, vmax &＃61; 1, center &＃61; 0)
# vmin&＃61;0, vmax&＃61;1 : the scope of colorbar value
# center&＃61;0 : colorbar valuee centered at 0
ax &＃61; sns.heatmap(uniform_data)# data 2
flights_long &＃61; sns.load_dataset("flights")
flights &＃61; flights_long.pivot("month", "year", "passengers")
f, ax &＃61; plt.subplots(figsize &＃61; (9, 6))
# annot &＃61; True: show the value
sns.heatmap(flights, annot &＃61; True, fmt &＃61; &＃39;d&＃39;, cmap &＃61; &＃39;Y|GnBu&＃39;, linewidth &＃61; 5, ax &＃61; ax)
label_y &＃61; ax.get_yticklabels()
plt.setp(label_y, rotation &＃61; 360, horizontalalignment &＃61; &＃39;right&＃39;)
label_x &＃61; ax.get_xticklabels()
plt.setp(label_x, rotation &＃61; 45, horizontalalignment &＃61; &＃39;right&＃39;)plt.show()# https://www.jianshu.com/p/d105d934d876