基于GAN的3D重建项目：64×64×64 numpy数组可视化问题求助

你用plt.imshow报错的原因很直接：这个函数只支持2D灰度数组（形状为(H,W)）或带通道的3D数组（比如(H,W,3)的RGB图），而你的(64,64,64)是三维体数据，没有通道维度，不符合输入要求。

下面是几种实用的可视化方法：

1. 正交切片可视化 #

通过提取三个正交平面（XY、XZ、YZ）的切片，可以快速查看体数据的内部结构，适合快速验证结果：

import numpy as np
import matplotlib.pyplot as plt

# 你的3D体数据
array = np.random.rand(64, 64, 64)

# 创建子图展示三个正交平面的中间切片
fig, axes = plt.subplots(1, 3, figsize=(15, 5))

# XY平面（取z轴中间位置的切片）
axes[0].imshow(array[32, :, :], cmap='gray')
axes[0].set_title('XY平面（z=32）')
axes[0].axis('off')

# XZ平面（取y轴中间位置的切片）
axes[1].imshow(array[:, 32, :], cmap='gray')
axes[1].set_title('XZ平面（y=32）')
axes[1].axis('off')

# YZ平面（取x轴中间位置的切片）
axes[2].imshow(array[:, :, 32], cmap='gray')
axes[2].set_title('YZ平面（x=32）')
axes[2].axis('off')

plt.tight_layout()
plt.show()

如果需要交互式切换切片位置（适合Jupyter环境），可以用滑块控件：

import ipywidgets as widgets
from IPython.display import display

def show_slice(z):
    plt.imshow(array[z, :, :], cmap='gray')
    plt.title(f'XY平面 z={z}')
    plt.axis('off')
    plt.show()

# 创建可拖动的滑块
slider = widgets.IntSlider(min=0, max=63, value=32, description='Z轴切片:')
widgets.interact(show_slice, z=slider)

2. 3D体绘制 #

如果需要直观展示整个3D结构，可以用体渲染工具，比如Plotly的Volume组件（支持交互式旋转、缩放）：

import plotly.graph_objects as go

fig = go.Figure(data=go.Volume(
    x=np.arange(64),
    y=np.arange(64),
    z=np.arange(64),
    value=array,
    isomin=0.2,  # 控制透明度阈值，只显示强度在0.2-0.8之间的区域
    isomax=0.8,
    opacity=0.1,  # 整体透明度，值越小越通透
    surface_count=20,  # 生成的表面数量，越多细节越丰富
    colorscale='gray'
))

fig.update_layout(scene=dict(aspectmode='data'))  # 保持三维比例一致
fig.show()

内容的提问来源于stack exchange，提问作者Mihir Ravindra Patel