Как сохранить последнее изображение анимации matplotlib?
Я хотел бы знать, как сохранить последнее изображение FuncAnimation в Python 3. Вот код:
import matplotlib.animation
from mpl_toolkits.mplot3d import Axes3D
ani = matplotlib.animation.FuncAnimation(fig, _update_2d_graph, frames=9, interval=600, repeat_delay=5000, blit=False)
ani.save(output_dir, writer = "imagemagick")
где _update_2d_graph — функция, обеспечивающая различные смещения точек в зависимости от номера кадра:
def _update_2d_graph(num):
x_updated = np.hstack((pos_gt.x[:num].tolist(),pos_pred.x[:num].tolist()))
y_updated = np.hstack((pos_gt.y[:num].tolist(),pos_pred.y[:num].tolist()))
paths_gt = []
paths_ped = []
for i in range(num):
marker_gt = mmarkers.MarkerStyle('o')
marker_ped = mmarkers.MarkerStyle('x')
path_gt = marker_gt.get_path().transformed(marker_gt.get_transform())
path_ped = marker_ped.get_path().transformed(marker_ped.get_transform())
paths_gt.append(path_gt)
paths_ped.append(path_ped)
graph._paths = np.concatenate([paths_gt, paths_ped])
graph.set_facecolors(np.concatenate([colors[:num], colors[:num]]))
graph.set_edgecolors(np.concatenate([colors[:num], colors[:num]]))
graph.set_offsets(np.hstack((y_updated[:, np.newaxis], x_updated[:, np.newaxis])))
Вместо сохранения всей анимации (например, в виде файла .gif) я хочу сохранить .png последнего временного шага анимации. Можно ли получить доступ к последнему кадру в объекте анимации?
Я не хочу вызывать _update_2d_graph(9), так как этот вызов уже был выполнен объектом анимации, и я не хочу повторять все 9 кадров через другой вызов FuncAnimation.
@DizietAsahi Асахи готово
Я не понимаю, зачем запускать анимацию, чтобы она ничего не делала и сохраняла последний кадр. Если вы просто хотите сохранить последний кадр, вы должны удалить всю анимацию (ani = ....) и вызвать _update_2d_graph(8); fig.savefig(...), чтобы сохранить последний кадр.
Этот пример не показывает мой полный вариант использования. В моем полном случае использования я создал анимацию для хранения анимации в формате .gif. Моя идея заключалась в том, что, поскольку объект уже создан, в идеале снова использовать объект анимации для доступа к последнему кадру для хранения .png в дополнение к .gif. Думаю, я предполагал, что есть простой способ получить доступ к этой информации внутри объекта без повторной итерации.
Спасибо за указание на это. Я попытался уточнить свой вопрос.
FuncAnimation не хранит каждый отдельный кадр (это было бы довольно затратно по памяти, однако ArtistAnimation сделал бы это). В чем причина "Я не хочу вызывать _update_2d_graph(9)"? (также (9) не вызывается, (8) будет последним кадром, верно?)
мой плохой, это я написал 9 вместо 8 для последнего кадра
Ответы 1
На самом деле, мои первоначальные тесты показывают, что простое сохранение фигуры после запуска анимации сохраняет состояние последнего кадра. Можете ли вы проверить это, чтобы подтвердить?
ani = matplotlib.animation.FuncAnimation(fig, _update_2d_graph, frames=9, interval=600, repeat_delay=5000, blit=False)
ani.save(output_dir, writer = "imagemagick")
fig.savefig('last_frame.png')
Я использовал код из Пример анимированного трехмерного случайного блуждания:
import numpy as np
import matplotlib.pyplot as plt
import mpl_toolkits.mplot3d.axes3d as p3
import matplotlib.animation as animation
# Fixing random state for reproducibility
np.random.seed(19680801)
def Gen_RandLine(length, dims=2):
"""
Create a line using a random walk algorithm
length is the number of points for the line.
dims is the number of dimensions the line has.
"""
lineData = np.empty((dims, length))
lineData[:, 0] = np.random.rand(dims)
for index in range(1, length):
# scaling the random numbers by 0.1 so
# movement is small compared to position.
# subtraction by 0.5 is to change the range to [-0.5, 0.5]
# to allow a line to move backwards.
step = ((np.random.rand(dims) - 0.5) * 0.1)
lineData[:, index] = lineData[:, index - 1] + step
return lineData
def update_lines(num, dataLines, lines):
for line, data in zip(lines, dataLines):
# NOTE: there is no .set_data() for 3 dim data...
line.set_data(data[0:2, :num])
line.set_3d_properties(data[2, :num])
return lines
# Attaching 3D axis to the figure
fig = plt.figure()
ax = p3.Axes3D(fig)
# Fifty lines of random 3-D lines
data = [Gen_RandLine(25, 3) for index in range(50)]
# Creating fifty line objects.
# NOTE: Can't pass empty arrays into 3d version of plot()
lines = [ax.plot(dat[0, 0:1], dat[1, 0:1], dat[2, 0:1])[0] for dat in data]
# Setting the axes properties
ax.set_xlim3d([0.0, 1.0])
ax.set_xlabel('X')
ax.set_ylim3d([0.0, 1.0])
ax.set_ylabel('Y')
ax.set_zlim3d([0.0, 1.0])
ax.set_zlabel('Z')
ax.set_title('3D Test')
# Creating the Animation object
line_ani = animation.FuncAnimation(fig, update_lines, 25, fargs=(data, lines),
interval=50, blit=False, repeat=True)
### HERE ARE THE ONLY LINES I ADDED ####
line_ani.save('./test.gif', writer='imagemagick')
fig.savefig('./test.png')
файл test.gif
файл test.png
Пожалуйста, предоставьте код (или его упрощенную форму) для
_update_2d_graph, ответ, скорее всего, будет зависеть от того, как вы написали эту функцию. Это может быть так же просто, как запуститьupdate_2d_graph(9), чтобы получить последний кадр, или вам, возможно, придется перебрать весь кадр, чтобы получить окончательный результат.