Сохранение анимации scatterplot с помощью matplotlib создает пустой видеофайл
У меня очень похожая проблема с этим вопросом
но предложенное решение не работает для меня.
Я настроил анимированный график рассеяния, используя модуль анимации matplotlib. Это отлично работает, когда он отображается в прямом эфире. Я хотел бы сохранить его в файл AVI или что-то подобное. Код, который я написал для этого, не выдает ошибку, но видео, которое он создает, просто показывает пустой набор осей или черный экран. Я сделал несколько проверок, и данные запускаются и обновляются цифры, они просто не сохраняются в видео...
Я попытался удалить "animated=True" и "blit=True", как предложено в этом вопросе, но это не решило проблему.
Я поместил соответствующий код ниже, но могу предоставить больше при необходимости. Кто-нибудь может подсказать, что я должен сделать, чтобы это работало?
def initAnimation(self):
rs, cfgs = next(self.jumpingDataStreamIterator)
#self.scat = self.axAnimation.scatter(rs[0], rs[1], c=cfgs[0], marker='o')
self.scat = self.axAnimation.scatter(rs[0], rs[1], c=cfgs[0], marker='o', animated=True)
return self.scat,
def updateAnimation(self, i):
"""Update the scatter plot."""
rs, cfgs = next(self.jumpingDataStreamIterator)
# Set x and y data...
self.scat.set_offsets(rs[:2,].transpose())
#self.scat = self.axAnimation.scatter(rs[0], rs[1], c=cfgs[0], animated=True)
# Set sizes...
#self.scat._sizes = 300 * abs(data[2])**1.5 + 100
# Set colors..
#self.scat.set_array(cfgs[0])
# We need to return the updated artist for FuncAnimation to draw..
# Note that it expects a sequence of artists, thus the trailing comma.
matplotlib.pyplot.draw()
return self.scat,
def animate2d(self, steps=None, showEvery=50, size = 25):
self.figAnimation, self.axAnimation = matplotlib.pyplot.subplots()
self.axAnimation.set_aspect("equal")
self.axAnimation.axis([-size, size, -size, size])
self.jumpingDataStreamIterator = self.jumpingDataStream(showEvery)
self.univeseAnimation = matplotlib.animation.FuncAnimation(self.figAnimation,
self.updateAnimation, init_func=self.initAnimation,
blit=True)
matplotlib.pyplot.show()
def animate2dVideo(self,fileName=None, steps=10000, showEvery=50, size=25):
self.figAnimation, self.axAnimation = matplotlib.pyplot.subplots()
self.axAnimation.set_aspect("equal")
self.axAnimation.axis([-size, size, -size, size])
self.Writer = matplotlib.animation.writers['ffmpeg']
self.writer = self.Writer(fps=1, metadata=dict(artist='Universe Simulation'))
self.jumpingDataStreamIterator = self.jumpingDataStream(showEvery)
self.universeAnimation = matplotlib.animation.FuncAnimation(self.figAnimation,
self.updateAnimation, scipy.arange(1, 25), init_func=self.initAnimation)
self.universeAnimation.save('C:/universeAnimation.mp4', writer = self.writer)
1 ответ
Извините за задержку. Я нашел обходной путь, просто сохранив множество отдельных изображений, а затем вызвав ffmpeg, чтобы связать их вместе. Это не идеально, но делает работу. (часть большего класса)
def easyway(self, frames):
##INSERT CODE TO GET xdata and ydata!
for i in range(0, frames):
fileName = "videoName"+"%04d.png" % i
matplotlib.pyplot.scatter(xdata,ydata,c=coldata, marker='*',
s=15.0, edgecolor='none')
matplotlib.pyplot.savefig(self.workingDirectory+'temp/'+fileName, dpi=dpi)
matplotlib.pyplot.close()
##INSERT CODE TO UPDATE xdata and ydata!
self.createVideoFile(fps)#calls FFMpeg to chain together all the pictures
if cleanUp:#removes all the picture files created in the process
print "temporary picture files being removed..."
self.clearDirectory()
print "FINISHED"
def clearDirectory(self,directoryName='temp'):
files = glob.glob(self.workingDirectory+directoryName+"/*")
for f in files:
os.remove(f)
def createVideoFile(self,fps=3):
command="ffmpeg -f image2 -r %s -i %s%s" % (str(fps), self.workingDirectory+'temp/', self.universe.description)
command+="%04d.png"
command+=" -c:v libx264 -r 30 %s.mp4" % (self.workingDirectory+'videos/'+self.universe.description)
print "Running command:"
print command
p = subprocess.Popen(command, shell=True, stdout = subprocess.PIPE, stderr=subprocess.STDOUT)
output = p.communicate()[0]
print "output\n"+"*"*10+"\n"
print output
print "*"*10
print "Video file has been written"