绘制3D热力图遇形状不匹配错误求助

问题背景 #

拥有形状为(100,519,492)的3D数组，需要绘制3D热力图：

• 用数组的索引（z轴对应第几个2D切片，y轴对应行索引，x轴对应列索引）定义点的位置
• 用数组的值定义点的颜色（0为白色、1为红色）
• 本质是将2D数据集重复绘制100次，形成按数值着色的立体区域

运行代码时触发错误：

ValueError: shape mismatch: objects cannot be broadcast to a single shape.  Mismatch is between arg 0 with shape (492,) and arg 1 with shape (519,).

错误原因 #

原代码中x = np.arange(0,492)、y = np.arange(0,519)、z = np.arange(0,100)都是一维数组，长度分别为492、519、100。而ax.scatter()要求输入的x、y、z必须是长度完全相同的一维数组（每个元素对应一个点的三维坐标），三者长度不匹配导致无法广播，触发形状错误。此外原代码未使用数组值设置颜色，不符合需求。

修正后的代码 #

from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
import numpy as np
from matplotlib import cm
from matplotlib.colors import Normalize

# 假设ODS_diff是形状为(100,519,492)的3D数组，若已有真实数据可替换此行
ODS_diff = np.random.rand(100, 519, 492)

# 生成三维坐标网格：z对应切片索引，y对应行，x对应列
z, y, x = np.meshgrid(np.arange(100), np.arange(519), np.arange(492), indexing='ij')

# 将网格和颜色数组都展开为一维数组（每个点对应一个坐标和颜色值）
x_flat = x.ravel()
y_flat = y.ravel()
z_flat = z.ravel()
color_flat = ODS_diff.ravel()

# 创建画布和3D轴
fig = plt.figure(figsize=(12, 10))
ax = fig.add_subplot(111, projection='3d')

# 设置颜色映射：0对应白色，1对应红色
norm = Normalize(vmin=0, vmax=1)
scatter = ax.scatter(x_flat, y_flat, z_flat, marker='s', s=10, 
                     c=color_flat, cmap=cm.Reds_r, norm=norm)

# 添加颜色条
plt.colorbar(scatter, label='数值')

# 设置标题和坐标轴标签
ax.set_title('3D热力图')
ax.set_xlabel('X轴（列索引）')
ax.set_ylabel('Y轴（行索引）')
ax.set_zlabel('Z轴（切片索引）')

plt.show()

代码说明 #

1. 生成三维坐标网格：使用np.meshgrid并设置indexing='ij'，确保z对应数组的第一维度（切片数）、y对应第二维度（行数）、x对应第三维度（列数），生成的网格形状与原3D数组一致。
2. 展开数组：用ravel()将三维网格和颜色数组转为一维，保证x、y、z、color的长度完全匹配，满足scatter的输入要求。
3. 颜色映射设置：通过Normalize固定数值范围0-1，搭配cm.Reds_r实现0为白色、1为红色的着色效果。
4. 优化显示：调整marker='s'的大小s=10避免点重叠，添加颜色条标注数值含义。

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