Python3.6.3下vtk_show致IPython崩溃，求适配Python3的解决方案

我之前也碰到过一模一样的问题——在IPython里调用vtk_show弹出外部窗口直接导致控制台崩溃，查了好久才发现是IPython没法处理这种外部渲染窗口的交互。你找到的Python2方案改改就能在Python3.6.3上用，核心是把字符串IO的部分适配成Python3的字节流处理：

先看修改后的完整代码：

import vtk
from IPython.display import Image
import io  # Python3需要导入io模块处理字节流

def vtk_show(renderer, width=400, height=300):
    """ Takes vtkRenderer instance and returns an IPython Image with the rendering. """
    # 创建vtk渲染窗口和交互器
    render_window = vtk.vtkRenderWindow()
    render_window.SetSize(width, height)
    render_window.AddRenderer(renderer)
    
    # 创建图像捕获对象
    window_to_image_filter = vtk.vtkWindowToImageFilter()
    window_to_image_filter.SetInput(render_window)
    window_to_image_filter.Update()
    
    # 把渲染结果写入字节流（Python3用BytesIO替代Python2的StringIO）
    writer = vtk.vtkPNGWriter()
    stream = io.BytesIO()
    writer.SetWriteToMemory(1)
    writer.SetInputConnection(window_to_image_filter.GetOutputPort())
    writer.Write()
    data = memoryview(writer.GetResult()).tobytes()
    stream.write(data)
    
    # 重置流指针并返回IPython可显示的Image对象
    stream.seek(0)
    return Image(stream.read())

关键改动点说明： #

• 导入io模块：Python3中不再使用StringIO.StringIO，而是用io.BytesIO来处理二进制字节流，这是Python2和3在IO处理上的核心差异之一。
• 内存写入与字节转换：vtk的vtkPNGWriter在Python3中返回的内存数据需要用memoryview(...).tobytes()转换成字节串，才能正确写入BytesIO流。
• 移除Python2的编码处理：原来Python2的代码可能涉及字符串编码转换，Python3中直接处理字节流就可以避免编码错误。

使用方法 #

你只需要像原来一样创建vtkRenderer对象，然后调用这个修改后的vtk_show函数，它会直接在IPython的单元格里渲染出图片，而不会弹出外部窗口导致崩溃：

# 示例：创建一个简单的立方体渲染器
renderer = vtk.vtkRenderer()
cube = vtk.vtkCubeSource()
mapper = vtk.vtkPolyDataMapper()
mapper.SetInputConnection(cube.GetOutputPort())
actor = vtk.vtkActor()
actor.SetMapper(mapper)
renderer.AddActor(actor)
renderer.SetBackground(0.1, 0.2, 0.3)

# 在IPython中显示渲染结果
vtk_show(renderer)

这样就完全适配Python3.6.3了，不会再出现控制台崩溃的问题，所有渲染结果都会直接内嵌在IPython界面里显示~

内容的提问来源于stack exchange，提问作者tomas-silveira