Matplotlib对数表面图的对数颜色映射异常问题求解

问题根源 #

1. 你的Z数据包含负数，直接执行np.log10(Z)会生成无效的nan值，导致颜色映射无法正常计算，出现单一颜色。
2. LogNorm仅适用于正数的对数归一化，且你传入plot_surface的是取log后的Z值，但norm的vmin/vmax用的是原始Z的范围，两者不匹配，进一步导致颜色映射失效。

解决方案 #

根据负数是否为有效数据，提供两种处理方式：

方式一：负数为噪声/无效值，仅关注绝对值

如果负数是测量误差或无意义数据，可取绝对值后处理，确保对数计算有效：

import matplotlib.pyplot as plt
import numpy as np
import matplotlib.colors as colors
import matplotlib.ticker as mticker

fig = plt.figure(figsize=(10,10))
ax = plt.axes(projection='3d')

def log_tick_formatter(val, pos=None):
    return "{:.2e}".format(10**val)

# 定义原始数据
X = np.arange(0,2,1)
Y = np.arange(0,3,1)
X,Y = np.meshgrid(X,Y)
Z = [[1.2e-11,1.3e-11,-1.8e-11],[6e-13,1.3e-13,2e-15]]

# 处理Z的绝对值，避免对数计算报错
Z_abs = np.abs(Z)
Z_log = np.log10(Z_abs)

# 设置Z轴刻度格式为原始数值
ax.zaxis.set_major_formatter(mticker.FuncFormatter(log_tick_formatter))
ax.set_zlim(np.log10(np.min(Z_abs)), np.log10(np.max(Z_abs)))

# 用原始绝对值做对数归一化，确保颜色映射匹配数值范围
norm = colors.LogNorm(vmin=np.min(Z_abs), vmax=np.max(Z_abs))
# 基于原始绝对值生成颜色，关联对数归一化规则
surf = ax.plot_surface(X, Y, Z_log, facecolors=plt.cm.rainbow(norm(Z_abs)), cmap='rainbow')

# 添加对应原始数值范围的颜色条
fig.colorbar(surf, shrink=0.5, aspect=5, norm=norm, ticks=[1e-15, 1e-13, 1e-11])
plt.show()

方式二：负数为有效数据，使用对称对数归一化

如果负数有实际意义，需要保留正负信息，可使用SymLogNorm处理跨越0的数值（0附近用线性刻度，其余区域用对数刻度）：

import matplotlib.pyplot as plt
import numpy as np
import matplotlib.colors as colors
import matplotlib.ticker as mticker

fig = plt.figure(figsize=(10,10))
ax = plt.axes(projection='3d')

# 定义原始数据
X = np.arange(0,2,1)
Y = np.arange(0,3,1)
X,Y = np.meshgrid(X,Y)
Z = [[1.2e-11,1.3e-11,-1.8e-11],[6e-13,1.3e-13,2e-15]]

# 自定义对称对数刻度格式，显示正负号和原始数值
def symlog_tick_formatter(val, pos=None):
    if val == 0:
        return "0"
    sign = "-" if val < 0 else ""
    return f"{sign}{:.2e}".format(np.abs(val))

# 对称对数归一化，linthresh设置线性区域的阈值（接近0的范围用线性）
norm = colors.SymLogNorm(linthresh=1e-14, vmin=np.min(Z), vmax=np.max(Z))
surf = ax.plot_surface(X, Y, Z, norm=norm, cmap='rainbow')

# 设置Z轴刻度格式
ax.zaxis.set_major_formatter(mticker.FuncFormatter(symlog_tick_formatter))

# 添加对应对称对数范围的颜色条
fig.colorbar(surf, shrink=0.5, aspect=5, norm=norm)
plt.show()

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