如何在Matplotlib的imshow()中添加图例?

如何在Matplotlib的imshow()中添加图例?

要在Matplotlib的imshow()中添加图例,可以按照以下步骤进行 –

  • 设置图形大小并调整子图之间和周围的填充。
  • 使用numpy创建随机数据。
  • 初始化一个颜色映射。
  • 从样本数据(步骤2)中获取唯一的数据点。
  • 使用不同的标签和颜色绘制每个颜色,以放置在图例中。
  • 将图例放在框的右上角。
  • 要显示图形,请使用show()方法。

例子

import numpy as np
from matplotlib import pyplot as plt, cm

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

data = np.random.rand(3, 3)
cmap = cm.YlOrBr
unique_data = np.unique(data)
i = 0

for entry in unique_data:
   mycolor = cmap(entry * 255 / (max(unique_data) - min(unique_data)))
   plt.plot(0, 0, "-", color=mycolor, label="%d"%i)
   i += 1

plt.imshow(data, cmap=cmap)
plt.legend(loc="upper right", bbox_to_anchor=(1.25, 1.0))

plt.show()

输出

如何在Matplotlib的imshow()中添加图例?

Python教程

Java教程

Web教程

数据库教程

图形图像教程

大数据教程

开发工具教程

计算机教程