为Matplotlib FuncAnimation添加全局固定范围的单一颜色条

要解决这个问题，核心是只创建一次颜色条，同时让每帧等高线图使用固定的颜色映射范围，避免重绘时清除或重复生成颜色条。下面是修改后的完整代码，以及关键步骤的解释：

修改后的完整可运行代码 #

import numpy as np
import itertools
import matplotlib.pyplot as plt
import matplotlib.mlab as ml
import matplotlib.animation as animation

def f(x,y,a):
    return a*(x**2+y**2)

# Generate input data
avals = list(np.linspace(0,1,10))
xaxis = list(np.linspace(-2,2,9))
yaxis = list(np.linspace(-2,2,9))
xy = list(itertools.product(xaxis,yaxis))
xy = np.array(xy)
x = xy[:,0]
y = xy[:,1]

# Prepare z values for all animation frames
zlist = []
for a in avals:
    z = [f(x[i], y[i], a) for i in range(len(x))]
    zlist.append(z)

# Calculate global min/max of z values to fix colorbar range
z_all = np.concatenate(zlist)
vmin, vmax = z_all.min(), z_all.max()

xi = np.linspace(min(x), max(x), len(x))
yi = np.linspace(min(y), max(y), len(y))

fig, ax = plt.subplots()

# Initialize first contour plot and create colorbar ONLY ONCE
zi_initial = ml.griddata(x, y, zlist[0], xi, yi, interp='linear')
contourplot = ax.contourf(xi, yi, zi_initial, cmap=plt.cm.hsv, origin='lower', vmin=vmin, vmax=vmax)
cbar = fig.colorbar(contourplot, ax=ax)
cbar.set_label('z Value')  # Optional: Add a label to the colorbar

def animate(index):
    # Remove previous contour collections (instead of clearing entire axis)
    for coll in ax.collections:
        coll.remove()
    
    # Generate data for current frame
    zi = ml.griddata(x, y, zlist[index], xi, yi, interp='linear')
    # Draw new contour with fixed color range
    ax.contourf(xi, yi, zi, cmap=plt.cm.hsv, origin='lower', vmin=vmin, vmax=vmax)
    ax.set_title(f'Frame {index:03d}')
    
    return ax.collections  # Return elements to update (optimizes blit if enabled)

# Create and run the animation
ani = animation.FuncAnimation(fig, animate, frames=len(avals), interval=200, blit=False)
plt.show()

关键修改说明 #

1. 固定颜色映射范围：
   先计算所有帧中z值的全局最小值vmin和最大值vmax，确保每帧等高线图使用完全一致的颜色刻度，颜色条范围不会随帧变化。

2. 只创建一次颜色条：
   在动画初始化阶段（animate函数外）绘制第一帧的等高线图，基于它创建颜色条。之后不再重复生成颜色条，彻底避免重复绘制导致的布局混乱。

3. 避免清除整个坐标轴：
   不再使用ax.clear()（会清除包括颜色条在内的所有轴元素），而是遍历并移除轴上的等高线集合（ax.collections），只更新等高线部分，保留颜色条和轴的其他元素。

4. 保持颜色一致性：
   每次调用contourf时都指定vmin和vmax参数，确保每帧的颜色映射与颜色条完全匹配，视觉效果统一。

这样修改后，动画过程中颜色条会稳定显示在图中，范围固定，不会出现重复绘制或混乱的问题。

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