基于Matplotlib实现.mha格式脑MRI图像切片滚动浏览的技术问询

我刚好做过类似的需求，给你一套亲测可行的实现步骤，用Matplotlib结合SimpleITK搞定.mha格式的MRI切片滚动：

第一步：准备依赖库 #

首先得安装读取.mha需要的SimpleITK，还有必备的matplotlib和numpy，用pip安装就行：

pip install simpleitk matplotlib numpy

第二步：读取.mha文件并转成NumPy数组 #

SimpleITK是处理医学图像的利器，读取.mha非常方便，之后转成NumPy数组才能用Matplotlib显示：

import SimpleITK as sitk
import numpy as np
import matplotlib.pyplot as plt

# 读取.mha文件
image = sitk.ReadImage("你的MRI文件路径.mha")
# 转成NumPy数组，注意SimpleITK默认维度是(z, y, x)，切片数在前
image_array = sitk.GetArrayFromImage(image)
# 获取总切片数
num_slices = image_array.shape[0]

第三步：实现交互式切片滚动 #

这里给你两种常用的交互方式，你可以选自己喜欢的：

方式一：用键盘左右箭头切换切片 #

通过Matplotlib的按键事件绑定切换逻辑，按下左箭头切上一个切片，右箭头切下一个，还会显示当前切片索引：

# 初始化当前切片索引
current_slice = 0

# 创建绘图窗口
fig, ax = plt.subplots()
img = ax.imshow(image_array[current_slice], cmap="gray")  # MRI用灰度图显示最合适
ax.set_title(f"Slice {current_slice + 1}/{num_slices}")

# 定义按键回调函数
def on_key_press(event):
    global current_slice
    if event.key == "right":
        current_slice = min(current_slice + 1, num_slices - 1)
    elif event.key == "left":
        current_slice = max(current_slice - 1, 0)
    # 更新图像和标题
    img.set_data(image_array[current_slice])
    ax.set_title(f"Slice {current_slice + 1}/{num_slices}")
    fig.canvas.draw_idle()

# 绑定按键事件
fig.canvas.mpl_connect("key_press_event", on_key_press)

plt.show()

方式二：用滑块控件拖动切换切片 #

如果觉得按键麻烦，也可以加一个滑块，拖动就能直观切换切片：

from matplotlib.widgets import Slider

# 创建主绘图区和滑块区域
fig, ax = plt.subplots()
plt.subplots_adjust(bottom=0.2)  # 给滑块留空间
img = ax.imshow(image_array[0], cmap="gray")
ax.set_title(f"Slice 1/{num_slices}")

# 创建滑块
ax_slider = plt.axes([0.2, 0.05, 0.6, 0.03])
slice_slider = Slider(ax_slider, "Slice", 0, num_slices - 1, valinit=0, valstep=1)

# 滑块回调函数
def update_slider(val):
    current_slice = int(val)
    img.set_data(image_array[current_slice])
    ax.set_title(f"Slice {current_slice + 1}/{num_slices}")
    fig.canvas.draw_idle()

slice_slider.on_changed(update_slider)

plt.show()

一些注意事项 #

• 如果发现切片显示方向不对，可以试试对数组做转置，比如image_array[current_slice].T，不同MRI数据的存储方向可能有差异。
• 灰度图的cmap="gray"是MRI显示的标准选择，你也可以试试cmap="bone"这类医学图像专用配色。

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