如何在Matplotlib中修改PDE解绘图坐标轴为0-10（兼容动画）

推荐方法：直接指定pcolormesh的坐标范围 #

这是最简洁高效的方式，不需要额外处理坐标轴，直接在绘制时传入对应0-10范围的网格边界即可。由于你的PDE解是20×20矩阵，对应的网格边界需要21个点（20个单元格对应21个边界点）。

代码示例（包含动画实现）：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation

# 模拟20×20的PDE初始解
current_sol = np.random.rand(20, 20)

# 创建0到10的网格边界数组
x_edges = np.linspace(0, 10, 21)
y_edges = np.linspace(0, 10, 21)

# 绘制初始热力图，指定坐标范围
pcm = plt.pcolormesh(x_edges, y_edges, current_sol)
plt.colorbar(label='PDE Solution')

# 动画更新函数：模拟每帧的PDE解变化
def update(frame):
    # 这里替换为你的PDE求解逻辑
    new_sol = np.random.rand(20, 20)
    pcm.set_array(new_sol.ravel())
    return [pcm]

# 创建动画
ani = FuncAnimation(plt.gcf(), update, frames=20, interval=300, blit=True)

plt.xlabel('X Axis')
plt.ylabel('Y Axis')
plt.show()

这样绘制后，坐标轴会自动显示0-10的范围，完全符合需求，且动画更新只需要修改热力图的数据数组，逻辑清晰。

备选方法：添加新坐标轴并隐藏原轴 #

如果必须保留原有不指定坐标的pcolormesh调用方式，可以通过以下步骤实现：

1. 获取当前坐标轴并隐藏
2. 创建与原轴位置重合的新坐标轴，设置0-10的范围

代码示例：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation

# 模拟20×20的PDE初始解
current_sol = np.random.rand(20, 20)

# 绘制初始热力图（默认坐标轴0-19）
pcm = plt.pcolormesh(current_sol)
plt.colorbar(label='PDE Solution')

# 隐藏原坐标轴
orig_ax = plt.gca()
orig_ax.set_visible(False)

# 创建新坐标轴，位置与原轴完全重合
new_ax = plt.axes(orig_ax.get_position())
new_ax.set_xlim(0, 10)
new_ax.set_ylim(0, 10)
new_ax.set_xlabel('X Axis')
new_ax.set_ylabel('Y Axis')

# 动画更新函数
def update(frame):
    new_sol = np.random.rand(20, 20)
    pcm.set_array(new_sol.ravel())
    return [pcm]

ani = FuncAnimation(plt.gcf(), update, frames=20, interval=300, blit=True)

plt.show()

注意：这种方法中，原热力图仍基于0-19的坐标绘制，新轴仅做刻度映射，不如推荐方法直接直观。

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