python - 绘制二维 numpy 数组的图例

Question

我喜欢使用 Numpy 和 Matplotlib 根据某些数字创建图例，但无济于事。因此，在将其转移到我的主脚本之前，我开始尝试使用测试函数以使其正确。

%matplotlib inline
import matplotlib.pyplot as plt
import numpy as np

min_xyz = np.random.randint(5, size=(50,1,50))

fig = plt.figure(figsize=(7,7))
ax = fig.add_subplot(111)
ax.set_title('test')
plt.imshow(min_xyz[:,0,:])
ax.set_aspect('equal')

ax.set_xlabel('Distance')
ax.set_ylabel('Depth')
ax.legend()

所以这会创建这样的东西

我想创建显示以下内容的图例:

 Mineral 1 = colour_1
 Mineral 2 = colour_2
 Mineral 3 = colour_3
 Mineral 4 = colour_4
 Mineral 5 = colour_5

我尝试使用 ax.legend() 但我似乎无法正确处理。有什么想法吗？

编辑:带有垂直彩条的解决方案

我使用 j08lue 输入的解决方案 - 垂直颜色条

%matplotlib inline
import matplotlib.pyplot as plt
import numpy as np
import matplotlib.patches as mpatches

min_xyz = np.random.randint(5, size=(50,1,50))

fig, ax = plt.subplots(figsize = (7,40))
ax.set_title('test')
cax = ax.imshow(min_xyz[:,0,:], cmap = plt.cm.Accent)

"""Handles for min_xyz"""
cbar = plt.colorbar(cax, ticks=[0, 1, 2, 3, 4], 
                    orientation='vertical',
                    fraction=0.045, pad=0.05)
cbar.ax.set_yticklabels(['Mineral 1', 'Mineral 2', 'Mineral 3', 'Mineral 4','Mineral 5'])

ax.set_xlabel('Distance')
ax.set_ylabel('Depth')

编辑:创建自定义图例

我已经放置了 j08lue 建议的解决方案，并设法使第一个图例正确。然而，我相信这与颜色条的标准化有关，以使图例反射(reflect)正确的颜色。我知道我错过了一些东西，但我不确定我应该寻找什么。任何意见都将不胜感激。

%matplotlib inline
import matplotlib.pyplot as plt
import numpy as np
import matplotlib.patches as mpatches

min_xyz = np.random.randint(5, size=(50,50))

fig2, ax2 = plt.subplots(figsize = (7,40))
ax = fig.add_subplot(111)
ax2.set_title('test')
cax2 = ax2.imshow(min_xyz, cmap = plt.cm.Accent, vmin=0, vmax=4)
ax2.set_aspect('equal')

"""Handles for min_xyz"""
 my_colors = {
    'Mineral 1' : 0.,
    'Mineral 2' : 1., # It is normalised to 0 to 1
    'Mineral 3' : 2.,
    'Mineral 4' : 3.,
    'Mineral 5' : 4.,
}
patches = [mpatches.Patch(color=cmap(v), label=k) for k,v in sorted(my_colors.items(), key=lambda t: t[0])]
plt.legend(handles=patches, loc=2, bbox_to_anchor=(1.01,1))

ax2.set_xlabel('Distance')
ax2.set_ylabel('Depth')

Answer 1

代理艺术家

这可以通过proxy artists来完成。文档中的示例:

import matplotlib.patches as mpatches
import matplotlib.pyplot as plt

red_patch = mpatches.Patch(color='red', label='The red data')
plt.legend(handles=[red_patch])

plt.show()

但是您需要弄清楚哪些颜色对应哪些值。例如

cmap = plt.cm.viridis
my_colors = {
    'Mineral 1' : 0.1,
    'Mineral 2' : 0.2,
    }

patches = [mpatches.Patch(color=cmap(v), label=k) for k,v in my_colors.items()]

plt.legend(handles=patches)

字典中的数字对应于标准化为 [0,1] 的数据，当然，您需要使用相同的 cmap 绘制数据。

替代方案:颜色条

或者，您可以添加 colorbar (相当于 imshow 绘图等中的图例)和 place your labels on the ticks .

cbar = plt.colorbar(cax, ticks=list(my_colors.values()), orientation='horizontal')
cbar.ax.set_xticklabels(list(my_colors.keys()))