Python mayavi屏幕截图，带有透明度

我试图用玛亚维场景的透明度保存截图。基于此，我编写了一个脚本来保存该图，但结果是输出混乱。下面是最简单的工作示例：在mayavi中，我创建了一个包含两个球体的场景，并将其先保存为RGB，然后保存为RGBA格式。当RGBA文件一团糟时，保存到RGB可以工作。我相信，问题出在mayavi方面，因为如果我从mayavi拍摄RGB图像，添加透明通道并使用PIL保存文件，结果就是我所期望的

这是一个错误还是mayavi rgba格式应该以某种方式转换为PIL接受的格式

from mayavi import mlab
from PIL import Image

fig=mlab.figure(1, bgcolor=(1, 1, 1), size=(700, 700))
# Set camera position and properties
fig.scene.parallel_projection = True
fig.scene.show_axes = True

# Draw atoms
x, y, z, t = [0.0,1.0] , [0.0,1.0], [0.0,0.0], [1,2]
dat = mlab.pipeline.scalar_scatter(x, y, z, t, figure=fig)
fig = mlab.pipeline.glyph(dat,scale_mode='none', scale_factor=0.5, figure=fig)

imgmap_RGB = mlab.screenshot(figure=fig, mode='rgb', antialiased=True)
img_RGB = Image.fromarray(imgmap_RGB, 'RGB')
img_RGB.save('foo_RGB.png')

imgmap_RGBA = mlab.screenshot(figure=fig, mode='rgba', antialiased=True)
img_RGBA = Image.fromarray(imgmap_RGBA, 'RGBA')
img_RGBA.save('foo_RGBA.png')

mlab.show()

---

出于某种原因，我不知道，对于RGBA数据，mayavi将返回0到1之间的浮点值，对于RGB数据，将返回无符号整数，请参阅（我在文档中找不到该信息）

要转换，请将

img\u RGBA=…

行替换为

img_RGBA = Image.fromarray(np.array(imgmap_RGBA*255, dtype=np.uint8))

---

之后我可以成功地查看png文件。

谢谢。我意识到问题就在这里，但是转换不正确，简单地转换成int没有帮助。