python - 在颜色栏中结束刻度 - matplotlib

Question

我正在尝试使用 matplotlib 库在 Python 中绘制颜色条。情节运作良好，但彩条不会在彩条上显示结束刻度。

使用以下命令可以正确绘制颜色条，但我只有从 -1.6 到 +1.6 的刻度（顶部和底部刻度不存在）。我的数据范围是-2到+2。

fig.colorbar(surf, shrink=1, aspect=12)

如下图所示：

我尝试使用以下命令：

cbar.set_ticks([-2,-1,0,1,2]);

但这也未能得到正确的结果。

更新：

我尝试使用 ticks=[-2,0,2] 参数，但效果不佳。这是我的绘图代码片段：

from mpl_toolkits.mplot3d import Axes3D
from matplotlib import cm
from matplotlib.ticker import LinearLocator, FormatStrFormatter
import matplotlib.pyplot as plt
import numpy as np

fig = plt.figure()
ax = fig.gca(projection='3d')
CHI = np.linspace(-45, 45, 35);
M = np.linspace(0, 1, 35)
CHI, M = np.meshgrid(CHI, M)
R = 2*M*np.sin(  2 * np.deg2rad(CHI) )
surf = ax.plot_surface(CHI, M, R, rstride=1, cstride=1, cmap=cm.hsv,linewidth=0, antialiased=False)
ax.set_xlim(-45,45)
cbar = plt.colorbar(surf, shrink=1, aspect=12, ticks=[-2,-1,0,1,2])
plt.show()

这将产生以下图：

如图所示，颜色条缺少结束刻度，即。-2 和 2

Answer 1

from mpl_toolkits.mplot3d import Axes3D
from matplotlib import cm
from matplotlib.ticker import LinearLocator, FormatStrFormatter
import matplotlib.pyplot as plt
import numpy as np

fig = plt.figure()
ax = fig.gca(projection='3d')
CHI = np.linspace(-45, 45, 35);
M = np.linspace(0, 1, 35)
CHI, M = np.meshgrid(CHI, M)
R = 2*M*np.sin(  2 * np.deg2rad(CHI) )
surf = ax.plot_surface(CHI, M, R, rstride=1, cstride=1, cmap=cm.hsv,linewidth=0, antialiased=False)
surf.set_clim([-2, 2]) # <- this is the important line
ax.set_xlim(-45,45)
cbar = plt.colorbar(surf, shrink=1, aspect=12, ticks=[-2,-1,0,1,2])
plt.show()

如果您明确将 clim 设置为有效，则自动缩放出现问题（我猜浮点相等测试存在问题）[-2, 2]。

Answer 2

您应该能够ticks在调用时设置参数plt.colorbar：

import matplotlib.pyplot as plt
import numpy as np

fig, ax = plt.subplots()
cmap = plt.get_cmap('jet')
data = np.random.randint(-2,3, size=(10,10))
im = ax.imshow(data, interpolation='nearest', cmap=cmap)
cbar = plt.colorbar(im, ticks=[-2,-1,0,1,2])
plt.show()