plt.contour绘制地震波图：色标渐变与消除白色背景需求

我来帮你搞定这个绘图难题！结合你的描述和代码，咱们一步步解决白色背景、边界异常颜色，同时实现接近pcolormesh的平滑渐变效果：

问题根源分析 #

1. plt.contour的局限性：它本身是绘制等高线条，而非填充区域，低振幅区域自然会显示空白背景，这是它的特性——你需要用plt.contourf（填充等高线）才能实现区域着色。
2. 白色背景与边界异常：你用contourf后出现的问题，本质是等高线层级（levels）设置不足，导致部分振幅区间未被覆盖，或是极值拉高了色标范围，让低振幅区域显示默认的"空白/异常颜色"。

针对性解决方案 #

下面是修改后的代码，我会标注关键调整点：

import numpy as np
import matplotlib.pyplot as plt

# 你的原有数据处理逻辑
T = len(time[0])*(time[0][1] - time[0][0]) 
Z = abs(ampFFT)*(T) 
maxFreq = abs(freq).max()

# 关键调整1：用百分位数替代硬切最大值，避免极值干扰色标分布
# 95百分位数既能保留高振幅细节，又不会让少数峰值吃掉大部分色彩区间
maxAmpFFT = np.percentile(Z, 95)
minAmpFFT = Z.min()

# 关键调整2：生成足够多的等高线层级，确保覆盖所有振幅区间
# 200个层级能实现接近pcolormesh的平滑渐变效果
levels = np.linspace(minAmpFFT, maxAmpFFT, num=200)

plt.figure(1)
# 关键调整3：使用contourf并配置参数消除异常
contour_plot = plt.contourf(
    time, freq, Z,
    levels=levels,          # 用自定义层级覆盖所有振幅
    cmap='viridis',         # 选择清晰的配色（也可用地震领域常用的'jet'）
    vmin=minAmpFFT,
    vmax=maxAmpFFT,
    extend='neither'        # 关闭超出范围的异常颜色显示（因为层级已覆盖所有数据）
)

plt.colorbar(contour_plot)
plt.ylim(0, maxFreq)
plt.title("Amplitude intensity regarding to time and frequency")
plt.xlabel('time (in seconds)')
plt.ylabel('frequency (in Hz)')
plt.show()

额外优化建议 #

如果仍存在零星白色区域，大概率是数据中有NaN值，可在处理Z时加上：

# 将NaN替换为最小振幅，避免空白
Z = np.nan_to_num(Z, nan=minAmpFFT)

为什么这个方案能解决你的问题？ #

• 自定义足够多的levels保证了每个振幅值都能对应到色标区间，彻底消除白色空白；
• 用百分位数替代硬切最大值，避免少数高振幅峰值挤压大部分区域的色彩表现；
• extend='neither'关闭了超出vmin/vmax的异常颜色绘制，解决边界深蓝色问题。

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