python如何在图例上加点_python – 在我的热图图上添加一个图例

我正在尝试将数据从csv文件转换为热图.目前,我的热图如下所示：

但它错过了一个传奇！这是我的代码：

import matplotlib.pyplot as plt

import numpy as np

column_labels = list(range(0,24))

row_labels = ["Lundi",

"Mardi",

"Mercredi",

"Jeudi",

"Vendredi",

"Samedi",

"Dimanche"]

data = np.array([

[0,0,0,0,0,0,0,0,0,0,434,560,650,340,980,880,434,434,0,0,0,0,0,0],

[0,0,0,0,0,0,0,434,560,0,650,0,0,0,0,340,980,0,0,0,880,0,434,343],

[0,0,0,0,0,0,0,0,0,0,434,560,650,340,980,880,434,434,0,0,0,0,0,0],

[0,0,0,0,0,0,0,434,560,0,650,0,0,0,0,340,980,0,0,0,880,0,434,343],

[0,0,0,0,0,0,0,0,0,0,434,560,650,340,980,880,434,434,0,0,0,0,0,0],

[0,0,0,0,0,0,0,434,560,0,650,0,0,0,0,340,980,0,0,0,880,0,434,343],

[0,0,0,0,0,0,0,0,0,0,434,560,650,340,980,880,434,434,0,0,0,0,0,0]

])

fig, axis = plt.subplots() # il me semble que c'est une bonne habitude de faire supbplots

heatmap = axis.pcolor(data, cmap=plt.cm.Blues) # heatmap contient les valeurs

axis.set_yticks(np.arange(data.shape[0])+0.5, minor=False)

axis.set_xticks(np.arange(data.shape[1])+0.5, minor=False)

axis.invert_yaxis()

axis.set_yticklabels(row_labels, minor=False)

axis.set_xticklabels(column_labels, minor=False)

fig.set_size_inches(11.03, 3.5)

plt.savefig('test.png', dpi=100)

我怎么能有这个情节的传奇？就像在这一个：

非常感谢你的帮助 ！