Numpy meshgrid轴求和降维问题：多轴求和语法及错误排查

核心错误 #

• 维度不兼容：你用meshgrid(a,b,c)生成的A/B/C是3D数组，而meshgrid(d,e)生成的D/E是2D数组，直接把它们传给函数f时，numpy无法自动广播不同维度的数组，运算会直接报错。
• 求和轴搞反：meshgrid默认用xy索引，生成的A数组形状是(len(b), len(a), len(c))，第一个轴对应b的取值，第二个轴才是a的取值，所以sum(axis=0)是对b轴求和，不是你想要的a轴。
• 绘图输入维度错误：就算前面步骤正确，B是3D数组，直接传给pcolormesh会因维度不匹配出问题，需要提取对应维度的网格数据。

修正后的完整代码 #

import numpy as np
import matplotlib.pyplot as plt

a = np.linspace(-10, 10)
b = np.linspace(-10, 10)
c = np.linspace(-10, 10)
d = np.linspace(-5, 5)
e = np.linspace(-7, 7)

# 用indexing='ij'生成网格，让每个变量的轴顺序和输入1D数组对应：a→axis0，b→axis1，c→axis2，d→axis3，e→axis4
A, B, C, D, E = np.meshgrid(a, b, c, d, e, indexing='ij')

def f(a_val, b_val, c_val, d_val, e_val):
    return a_val**2 + b_val**3 + c_val**6 + d_val + 2*e_val**3

# 对a轴（axis0）求和，得到4D数组（b,c,d,e）
F = np.sum(f(A, B, C, D, E), axis=0)
# 对d（axis2）和e（axis3）轴求和，得到2D数组（b,c）
G = np.sum(F, axis=(2, 3))

# 提取b和c对应的2D网格（取固定的a/d/e切片，得到纯b-c网格）
B_mesh = B[0, :, :, 0, 0]
C_mesh = C[0, :, :, 0, 0]

plt.pcolormesh(B_mesh, C_mesh, G)
plt.colorbar()
plt.show()

内存优化版（避免生成高维数组） #

如果直接生成5D数组内存不足，可以分步计算各部分的和，利用广播减少内存占用：

import numpy as np
import matplotlib.pyplot as plt

a = np.linspace(-10, 10)
b = np.linspace(-10, 10)
c = np.linspace(-10, 10)
d = np.linspace(-5, 5)
e = np.linspace(-7, 7)

# 计算a项的总和（标量）
sum_a = np.sum(a**2)
# 生成b和c的2D网格，计算b+c项
B, C = np.meshgrid(b, c, indexing='ij')
bc_part = B**3 + C**6
# 计算d+e项的总和（标量）
sum_de = np.sum(d + 2*e**3)

# 合并所有部分得到最终2D数组
G = sum_a + bc_part + sum_de

plt.pcolormesh(B, C, G)
plt.colorbar()
plt.show()

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