使用 imageio 保存 3D 绘图答案

【问题标题】：Save 3D plot with imageio使用 imageio 保存 3D 绘图
【发布时间】：2021-08-10 20:33:58
【问题描述】：

我想将 3D 图保存为 gif。但是，由于我无法解释的原因，以下代码不起作用。我收到错误消息：Image must be 2D (grayscale, RGB, or RGBA)，当我使用 imageio.mimsave(...) 时，我将图像保存为 RGB：

import numpy as np
import matplotlib.pyplot as plt
import imageio

x = [0,1,0,0]
y = [0,0,1,0]
z = [0,0,0,1]

fig = plt.figure(figsize=(15,9))
ax = fig.add_subplot(projection='3d')
ax.scatter(x,y,z,s=1500)

images = []
for n in range(0, 100):
    if n >= 20: 
        ax.azim = ax.azim+1.1 
    fig.canvas.draw()
    image = np.frombuffer(fig.canvas.tostring_rgb(), dtype='uint8')
    images.append(image)
   
imageio.mimsave('test.gif', images)

【问题讨论】：

下面我的回答，是不是当场？

标签： python image numpy matplotlib python-imageio

【解决方案1】：

np.frombuffer 返回一个平面数组，它的形状(X*Y*3,)，相反imageio 需要一个 Y × X × 3（或者可能 × 4 RGBA 缓冲区）。你必须reshape你的image

In [43]: import numpy as np
    ...: import matplotlib.pyplot as plt
    ...: import imageio
    ...: 
    ...: x = [0,1,0,0]
    ...: y = [0,0,1,0]
    ...: z = [0,0,0,1]
    ...: 
    ...: fig = plt.figure(figsize=(5, 3)) # @100 dpi it's 500×300 pixels
    ...: ax = fig.add_subplot(projection='3d')
    ...: ax.scatter(x,y,z,s=500)
    ...: 
    ...: images = []
    ...: for n in range(0, 25):
    ...:     if n >= 15:
    ...:         ax.azim = ax.azim+1.1
    ...:     fig.canvas.draw()
    ...:     image = np.frombuffer(fig.canvas.tostring_rgb(), dtype='uint8')
    ...:     images.append(image.reshape(300, 500, 3)) ## Y×X
    ...: 
    ...: imageio.mimsave('test.gif', images)
    ...: 

In [44]: ls -l test.gif
-rw-r--r-- 1 boffi boffi 825988 May 21 19:46 test.gif

In [45]:

请注意，我已修改尺寸和帧数以具有更小的 GIF。

【讨论】：