如何在子图中生成带有图像的动画(matplotlib)

时间:2017-03-15 17:39:32

标签: python matplotlib jupyter subplot

是否可以在jupyter笔记本中制作成对图像?

有两个图像列表:

greys = io.imread_collection(path_greys)
grdTru= io.imread_collection(path_grdTru)

以下天真的代码无法生成动画:

for idx in range(1,900):
    plt.subplot(121)
    plt.imshow(greys[idx], interpolation='nearest', cmap=plt.cm.gray)
    plt.subplot(122)
    plt.imshow(grdTru[idx], interpolation='nearest', cmap=plt.cm.,vmin=0,vmax=3)
    plt.show()

(它生成一个子图列表)

顺便说一下,example found in matplotlib doc如果粘贴在笔记本中就会失败。

1 个答案:

答案 0 :(得分:2)

为了让the example在jupyter笔记本中工作,您需要包含

%matplotlib notebook

魔术指令。

import numpy as np
%matplotlib notebook
import matplotlib.pyplot as plt
import matplotlib.animation as animation

fig = plt.figure()


def f(x, y):
    return np.sin(x) + np.cos(y)

x = np.linspace(0, 2 * np.pi, 120)
y = np.linspace(0, 2 * np.pi, 100).reshape(-1, 1)

im = plt.imshow(f(x, y), animated=True)


def updatefig(*args):
    global x, y
    x += np.pi / 15.
    y += np.pi / 20.
    im.set_array(f(x, y))
    return im,

ani = animation.FuncAnimation(fig, updatefig, interval=50, blit=True)
plt.show()

然后,您可以轻松地将其调整为您的图像列表。

从matplotlib 2.1版开始,您还可以选择内联创建JavaScript动画。

from IPython.display import HTML
HTML(ani.to_jshtml())

完整示例:

import numpy as np
%matplotlib inline
import matplotlib.pyplot as plt
import matplotlib.animation as animation

def f(x, y):
    return np.sin(x) + np.cos(y)

x = np.linspace(0, 2 * np.pi, 120)
y = np.linspace(0, 2 * np.pi, 100).reshape(-1, 1)

im = plt.imshow(f(x, y), animated=True);


def updatefig(*args):
    global x, y
    x += np.pi / 15.
    y += np.pi / 20.
    im.set_array(f(x, y))
    return im,

ani = animation.FuncAnimation(fig, updatefig, interval=50, blit=True)

from IPython.display import HTML
HTML(ani.to_jshtml())