Matplotlib contourf对数色标刻度过多问题求助

问题背景 #

使用Python 3.9.12 + Matplotlib 3.5.2在Jupyter Notebook中绘制对数色标的等高线图时，遇到两个问题：

• 限制z范围在1e-3到0.2时，色标刻度间隔为0.0025，高值区域刻度过于密集，不符合对数刻度的正常显示（每个数量级10个间隔刻度）
• 扩展z范围到4个数量级（比如1e-4到0.2）时，触发报错：Locator attempting to generate 4444 ticks ([0.000325, ..., 1.0]), which exceeds Locator.MAXTICKS (1000).

原始代码如下：

#set up a 1d plot
fig,axes = plt.subplots(1,1,figsize=(9.0,8.0),sharex=True)
ax1 = axes
from matplotlib import cm, ticker


ymin=1.0
ymax=15
xmin=4.0
xmax=45


zzN = np.nan_to_num(zzN)
z = ma.masked_where((zzN<=1e-3)|(zzN>0.2), zzN)
nrcont=ax1.contourf(p1,q1,z,locator=ticker.LogLocator(), cmap='coolwarm',alpha=0.3)


cbar = fig.colorbar(nrcont)
cbar.ax.set_ylabel('joint PDF',labelpad=15)


ax1.set_xlim(xmin, xmax) #in pairs
ax1.set_ylim(ymin,ymax)
ax1.set_xlabel('phonon energy [keV]')
ax1.set_ylabel('charge energy [keV]')
ax1.grid(True)
ax1.yaxis.grid(True,which='minor',linestyle='--')
#ax1.legend(loc=2,prop={'size':22})

for axis in ['top','bottom','left','right']:
  ax1.spines[axis].set_linewidth(2)

plt.tight_layout()

plt.show()

解决方案 #

核心思路 #

问题根源在于LogLocator的默认自动刻度生成逻辑：未指定参数时，Matplotlib会尝试生成线性间隔的刻度而非对数间隔，导致刻度过多或显示异常。需要手动配置LogLocator参数，同时指定对数间隔的等高线层级，从根源避免问题。

修改后的代码 #

import numpy as np
import matplotlib.pyplot as plt
from matplotlib import cm, ticker
from numpy import ma

# 初始化绘图
fig, axes = plt.subplots(1, 1, figsize=(9.0, 8.0), sharex=True)
ax1 = axes

# 坐标轴范围设置
ymin = 1.0
ymax = 15
xmin = 4.0
xmax = 45

# 处理数据：扩展z范围到4个数量级（1e-4到0.2）
zzN = np.nan_to_num(zzN)
z = ma.masked_where((zzN <= 1e-4) | (zzN > 0.2), zzN)

# 生成对数间隔的等高线层级：从1e-4到0.2，每个数量级10个间隔，共41个点（对应40条等高线）
levels = np.logspace(np.log10(1e-4), np.log10(0.2), 41)

# 绘制等高线：指定层级+配置对数定位器
nrcont = ax1.contourf(p1, q1, z, levels=levels, 
                      locator=ticker.LogLocator(base=10, numticks=11),
                      cmap='coolwarm', alpha=0.3)

# 配置色标：强制应用对数刻度
cbar = fig.colorbar(nrcont)
cbar.ax.set_ylabel('joint PDF', labelpad=15)
cbar.locator = ticker.LogLocator(base=10, numticks=11)
cbar.update_ticks()

# 坐标轴细节设置
ax1.set_xlim(xmin, xmax)
ax1.set_ylim(ymin, ymax)
ax1.set_xlabel('phonon energy [keV]')
ax1.set_ylabel('charge energy [keV]')
ax1.grid(True)
ax1.yaxis.grid(True, which='minor', linestyle='--')

for axis in ['top', 'bottom', 'left', 'right']:
    ax1.spines[axis].set_linewidth(2)

plt.tight_layout()
plt.show()

关键说明 #

• np.logspace(np.log10(1e-4), np.log10(0.2), 41)：生成严格的对数间隔层级，确保每个数量级包含10个间隔，避免自动生成的密集线性刻度
• ticker.LogLocator(base=10, numticks=11)：明确指定对数底数为10，每个数量级显示11个刻度（包含首尾），对应10个间隔，符合对数刻度的正常显示逻辑
• cbar.locator + cbar.update_ticks()：强制色标应用配置好的对数刻度，避免色标与等高线刻度不一致的问题

内容的提问来源于stack exchange，提问作者villaa