绘制2D 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建议的解决方案，并设法让第一个传奇正确。但是，我认为它与颜色条的标准化有关，以使图例反映出正确的颜色。我知道我错过了一些东西，但我不确定我应该寻找什么。任何意见都非常感谢。

%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()

但是你需要弄清楚哪些颜色与哪些颜色相对应。 E.g。

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

或者，您可以添加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()))