保存Matplotlib动画 [英] Save Matplotlib Animation

问题描述

我正在尝试制作wave包的动画并将其另存为电影。除保存之外的所有内容都可以正常工作。你能告诉我我做错了吗？当进入 ani.save（'MovWave.mp4'）行时，他告诉我：

I am trying to make an Animation of a wave package and save it as a movie. Everything except the saving is working. Can you please tell me what I am doing wrong? When going into the line ani.save('MovWave.mp4') he tells me:

    writer = writers.list()[0]
IndexError: list index out of range

我当然尝试使用谷歌搜索，但我什至不知道这意味着什么。

I tried googling it of course, but I don't even know what it means.

更新：立即在控制台中调用 ffmpeg 。它说我已经安装了ffmpeg版本 0.10.7-6：0.10.7-0jon1〜precise 。我更新了代码并运行了程序，但是现在出现以下错误：

UPDATE: I can call ffmpeg in console now. It says I have ffmpeg version 0.10.7-6:0.10.7-0jon1~precise installed. I updated the code and ran the program, but now I get the following error:

Traceback (most recent call last):
  ani.save('MovWave.mpeg', writer="ffmpeg")
  writer.grab_frame()
  dpi=self.dpi)
  self.canvas.print_figure(*args, **kwargs)
  self.figure.dpi = origDPI
  self.dpi_scale_trans.clear().scale(dpi, dpi)
  self._mtx = np.identity(3)
  from numpy import eye
  File "<frozen importlib._bootstrap>", line 1609, in _handle_fromlist
UnicodeDecodeError: 'utf-8' codec can't decode byte 0xff in position 0: invalid start byte

更新2：显然，使用python 3.3作为版本时存在错误Doctorlove指出。我现在正尝试使用python 2.7代替。现在，它创建了一个mpeg文件，但是无法播放，并且只有大约150 kB。

Update 2: Apparently there is a bug when using python 3.3 as doctorlove pointed out. I am now trying to use python 2.7 instead. Now it creates an mpeg-file but it cannot be played and it is only about ~150 kB big.

更新3：我在Win7机器上尝试了完全相同的代码，它也可以在python 3.3中使用。但是我有同样的问题，我早些时候使用python 2.7。创建的mpeg文件无法播放，只有几百kB。

Update 3: Okay, so I tried the exact same code on my Win7 machine and it also works in python 3.3. But I have the same problem, I had earlier with python 2.7. The mpeg-file created cannot be played and is only a few hundred kB.

#! coding=utf-8
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation
import time
time.clock()

def FFT(x,y):
    X = (x[-1]-x[0])/len(y)
    f = np.linspace(-2*np.pi/X/2,2*np.pi/X/2,len(y))
    F = np.fft.fftshift(np.fft.fft(y))/np.sqrt(len(y))
    return(f,F)

def FUNCTION(k_0,dx,c,t):
    y = np.exp(1j*k_0*(x-c*t))*np.exp(-((x-c*t)/(2*dx))**2 )*(2/np.pi/dx**2)**(1/4)
    k,F = FFT((x-c*t),y)
    return(x,y,k,F)

#Parameter
N = 1000
x   = np.linspace(0,30,N)
k_0 = 5
dx  = 1
c   = 1

l = [k_0,c,dx]

fig = plt.figure("Moving Wavepackage and it's FFT")
sub1 = plt.subplot(211)
sub2 = plt.subplot(212)
sub2.set_xlim([-10,10])
sub1.set_title("Moving Wavepackage and it's FFT")
sub1.set_ylabel("$Re[\psi(x,t)]$")
sub1.set_xlabel("$t$")
sub2.set_ylabel("$Re[\psi(k_x,t)]$")
sub2.set_xlabel("$k_x$")


n = 50
t = np.linspace(0,30,n)
img = []
for i in range(n):
    x,y,k,F = FUNCTION(k_0,dx,c,t[i]) 

    img.append(plt.plot(x,np.real(y),color="red", axes=plt.subplot(211)))
    img.append(plt.plot(k,np.real(F),color="red", axes=plt.subplot(212)))

ani = animation.ArtistAnimation(fig, img, interval=20, blit=True, repeat_delay=0)

ani.save('MovWave.mpeg', writer="ffmpeg")

print(time.clock())
plt.show()

推荐答案

您在文本中提到了 mencoder ，但没有提到

You mentioned mencoder in your text, but not the code.

Matplotlib文档在mencoder 。 org / examples / animation / movie_demo.html rel = nofollow>演示：

Matplotlib docs has a check for mencoder in a demo:

not_found_msg = """
The mencoder command was not found;
mencoder is used by this script to make an avi file from a set of pngs.
It is typically not installed by default on linux distros because of
legal restrictions, but it is widely available.
"""

try:
    subprocess.check_call(['mencoder'])
except subprocess.CalledProcessError:
    print "mencoder command was found"
    pass # mencoder is found, but returns non-zero exit as expected
    # This is a quick and dirty check; it leaves some spurious output
    # for the user to puzzle over.
except OSError:
    print not_found_msg
    sys.exit("quitting\n")

这篇关于保存Matplotlib动画的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！