1

我有一个项目,我试图绘制整数类型的二维数组。这些值可以在 0 到 4 之间变化,并且可能并不全部存在。

我开始使用整数数组resr、列出的颜色图cmap基于离散间隔的颜色图索引来 norm开发我的绘图例程:

import matplotlib.pyplot as plt
import numpy as np
import matplotlib.colors as colors
from mpl_toolkits.axes_grid1 import make_axes_locatable

resr = np.array([[0,2],[3,4]],dtype=int)

norm = colors.BoundaryNorm(np.unique(resr), len(np.unique(resr))-1)
cmap1 = ['#7fc97f', '#ffff99', '#386cb0', '#f0027f']
cmap = colors.ListedColormap(cmap1) 

fig,ax = plt.subplots()
im = ax.imshow(resr, cmap=cmap,norm=norm)
divider = make_axes_locatable(ax)
cax = divider.append_axes("right", size="5%")
bounds = 0.5+np.arange(norm.N+1)
cb = plt.colorbar(im, cmap=cmap,norm=norm,boundaries=bounds,cax=cax)
tic_locs =1+np.arange(norm.N+1)
cb.set_ticks(tic_locs)
cb.ax.set_yticklabels(cmap1)
cb.ax.tick_params(labelsize=10)

返回预期结果:

在此处输入图像描述

当我更改resr为测试以确保绘图例程在其他情况下有效时:

resr = np.array([[0,1],[3,4]],dtype=int)

图像绘制正确,但颜色条的格式不正确:

在此处输入图像描述

为什么会这样?

4

0 回答 0