动画是一种高效的可视化工具,能够提升用户的吸引力和视觉体验,有助于以富有意义的方式呈现数据可视化。本文的主要介绍在Python中两种简单制作动图的方法。其中一种方法是使用matplotlib的Animations模块绘制动图,另一种方法是基于Pillow生成GIF动图。
python学习资料已打包好,需要的小伙伴可以戳这里【python资料】
1 Animations模块
Matplotlib的Animations模块提供了FuncAnimation和ArtistAnimation类来创建matplotlib绘图动画,FuncAnimation和ArtistAnimation都是Animation类的子类。它们的区别在于实现动画的方式和使用场景不同。FuncAnimation适用于根据时间更新图形状态的动画效果,且更加灵活和常用。而ArtistAnimation适用于将已有的静态图像序列组合成动画的效果。具体区别如下:
-
FuncAnimation
:FuncAnimation是基于函数的方法来创建动画的。它使用用户提供的一个或多个函数来更新图形的状态,并按照一定的时间间隔连续地调用这些函数,从而实现动画效果。用户需要定义一个更新函数,该函数在每个时间步长上更新图形对象的属性,然后FuncAnimation会根据用户指定的帧数、时间间隔等参数来自动计算动画的帧序列。这种方法适用于需要根据时间变化来更新图形状态的动画效果。 -
ArtistAnimation
:ArtistAnimation是基于静态图像的方法来创建动画的。它要求用户提供一系列的静态图像,称为艺术家对象。这些图像可以是通过Matplotlib创建的任何类型的可视化对象,例如Figure、Axes、Line2D等。用户需要将这些静态图像存储在一个列表中,然后通过ArtistAnimation来显示这些图像的序列。ArtistAnimation会按照用户指定的时间间隔逐帧地显示这些图像,从而实现动画效果。这种方法适用于已经有一系列静态图像需要组合成动画的场景。
本节将通过几个示例来介绍Animations模块的使用,所介绍的示例出自:gallery-animation。
1.1 FuncAnimation类
FuncAnimation构造函数的参数含义如下:
fig
:要绘制动画的Figure对象。func
:用于更新每一帧的函数,该函数接受一个参数frame,表示当前待绘制的数据帧。frames
:用于产生待绘制的数据,可以是整数、生成器函数或迭代器。init_func
:在绘制动画之前调用的初始化函数。fargs
:传递给func
函数的附加参数(可选)。save_count
:指定动画中缓存的帧数量(可选),默认为100。注意该参数用于确定最后生成动图和视频所用图像的数量。interval
:每一帧之间的时间间隔,以毫秒为单位,默认为200。repeat
:控制动画是否重复播放,默认为True。repeat_delay
:重复动画之间的延迟时间(以毫秒为单位),默认为0。blit
:指定是否使用blitting技术来进行绘制优化,默认为False。cache_frame_data
:指定是否缓存帧数据,默认为True。
示例-生成动态的正弦波动画
import itertools | |
import matplotlib.pyplot as plt | |
import numpy as np | |
import matplotlib.animation as animation | |
# 定义生成数据的函数 | |
def data_gen(max_range): | |
# 使用itertools.count()生成无限递增的计数器 | |
for cnt in itertools.count(): | |
# 当计数器超过最大范围时停止生成数据 | |
if cnt > max_range: | |
break | |
print(cnt) | |
# 计算时间t和对应的y值,使用np.sin()计算sin函数,np.exp()计算指数函数 | |
t = cnt / 10 | |
yield t, np.sin(2*np.pi*t) * np.exp(-t/10.) | |
# 初始化函数,设置坐标轴范围和清空数据 | |
def init(): | |
ax.set_ylim(-1.1, 1.1) | |
ax.set_xlim(0, 1) | |
del xdata[:] | |
del ydata[:] | |
line.set_data(xdata, ydata) | |
return line, | |
# 创建图形对象以及子图对象 | |
fig, ax = plt.subplots() | |
# 创建线条对象 | |
line, = ax.plot([], [], lw=2) | |
# 创建文本对象用于显示 x 和 y 值 | |
text = ax.text(0., 0., '', transform=ax.transAxes) | |
# 设置文本位置 | |
text.set_position((0.7, 0.95)) | |
# 将文本对象添加到图形中 | |
ax.add_artist(text) | |
ax.grid() | |
xdata, ydata = [], [] | |
# 更新函数,将新的数据添加到图形中 | |
def run(data): | |
# 获取传入的数据 | |
t, y = data | |
# 将时间和对应的y值添加到xdata和ydata中 | |
xdata.append(t) | |
ydata.append(y) | |
# 获取当前坐标轴的范围 | |
xmin, xmax = ax.get_xlim() | |
# 更新文本对象的值 | |
text.set_text('x = {:.2f}, y = {:.2f}'.format(t, y)) | |
# 如果时间t超过当前范围,更新坐标轴范围 | |
if t >= xmax: | |
ax.set_xlim(xmin, 2*xmax) | |
# 重绘图形 | |
ax.figure.canvas.draw() | |
# 更新线条的数据 | |
line.set_data(xdata, ydata) | |
return line, text | |
# 创建动画对象 | |
# fig:图形对象 | |
# run:更新函数,用于更新图形中的数据 | |
# data_gen(20):生成器函数,产生数据的最大范围为20 | |
# interval=100:每帧动画的时间间隔为100毫秒 | |
# init_func=init:初始化函数,用于设置图形的初始状态 | |
# repeat=True:动画重复播放 | |
ani = animation.FuncAnimation(fig, run, data_gen(20), interval=100, init_func=init, repeat=True) | |
# 显示图形 | |
plt.show() |
示例-创建动态散点图与折线图
import matplotlib.pyplot as plt | |
import numpy as np | |
import matplotlib.animation as animation | |
# 创建一个图形窗口和坐标轴 | |
fig, ax = plt.subplots() | |
# 创建时间数组 | |
t = np.linspace(0, 3, 50) | |
# 自由落体加速度 | |
g = -9.81 | |
# 初始速度 | |
v0 = 12 | |
# 计算高度 | |
z = g * t**2 / 2 + v0 * t | |
# 第二个初始速度 | |
v02 = 5 | |
# 计算第二个高度 | |
z2 = g * t**2 / 2 + v02 * t | |
# 创建散点图 | |
scat = ax.scatter(t[0], z[0], c="b", s=5, label=f'v0 = {v0} m/s') | |
# 创建线图 | |
line2 = ax.plot(t[0], z2[0], label=f'v0 = {v02} m/s')[0] | |
# 设置坐标轴范围和标签 | |
ax.set(xlim=[0, 3], ylim=[-4, 10], xlabel='Time [s]', ylabel='Z [m]') | |
# 添加图例 | |
ax.legend() | |
def update(frame): | |
x = t[:frame] | |
y = z[:frame] | |
# 更新散点图 | |
data = np.stack([x, y]).T | |
# 更新散点图中每个点的位置 | |
scat.set_offsets(data) | |
# 更新线图 | |
line2.set_xdata(t[:frame]) | |
line2.set_ydata(z2[:frame]) | |
return (scat, line2) | |
# 创建动画 | |
# frames为数值表示动画的总帧数,即每次更新参数传入当前帧号 | |
ani = animation.FuncAnimation(fig=fig, func=update, frames=40, interval=30) | |
# 显示图形 | |
plt.show() |
示例-贝叶斯更新动画
import math | |
import matplotlib.pyplot as plt | |
import numpy as np | |
from matplotlib.animation import FuncAnimation | |
# 定义分布概率密度函数 | |
def beta_pdf(x, a, b): | |
return (x**(a-1) * (1-x)**(b-1) * math.gamma(a + b) | |
/ (math.gamma(a) * math.gamma(b))) | |
# 更新分布类,用于更新动态图 | |
class UpdateDist: | |
def __init__(self, ax, prob=0.5): | |
self.success = 0 | |
self.prob = prob | |
self.line, = ax.plot([], [], 'k-') | |
self.x = np.linspace(0, 1, 200) | |
self.ax = ax | |
# 设置图形参数 | |
self.ax.set_xlim(0, 1) | |
self.ax.set_ylim(0, 10) | |
self.ax.grid(True) | |
# 这条竖直线代表了理论值,图中的分布应该趋近于这个值 | |
self.ax.axvline(prob, linestyle='--', color='black') | |
def __call__(self, i): | |
# 这样图形可以连续运行,我们只需不断观察过程的新实现 | |
if i == 0: | |
self.success = 0 | |
self.line.set_data([], []) | |
return self.line, | |
# 根据超过阈值与均匀选择来选择成功 | |
if np.random.rand() < self.prob: | |
self.success += 1 | |
y = beta_pdf(self.x, self.success + 1, (i - self.success) + 1) | |
self.line.set_data(self.x, y) | |
return self.line, | |
# 设置随机状态以便再现结果 | |
np.random.seed(0) | |
# 创建图形和坐标轴对象 | |
fig, ax = plt.subplots() | |
# 创建更新分布对象,并应该收敛到的理论值为0.7 | |
ud = UpdateDist(ax, prob=0.7) | |
# 创建动画对象 | |
anim = FuncAnimation(fig, ud, frames=100, interval=100, | |
blit=True, repeat_delay=1000) | |
# 显示动画 | |
plt.show() |
示例-模拟雨滴
import matplotlib.pyplot as plt | |
import numpy as np | |
from matplotlib.animation import FuncAnimation | |
# 设置随机种子以确保可复现性 | |
np.random.seed(0) | |
# 创建画布和坐标轴对象 | |
fig = plt.figure(figsize=(7, 7)) | |
# 在画布上添加一个坐标轴对象。 | |
# [0, 0, 1, 1]参数指定了坐标轴的位置和大小,分别表示左下角的 x 坐标、左下角的 y 坐标、宽度和高度。 | |
# frameon=False参数表示不显示坐标轴的边框 | |
ax = fig.add_axes([0, 0, 1, 1], frameon=False) | |
ax.set_xlim(0, 1), ax.set_xticks([]) | |
ax.set_ylim(0, 1), ax.set_yticks([]) | |
# 创建雨滴数据 | |
n_drops = 50 | |
rain_drops = np.zeros(n_drops, dtype=[('position', float, (2,)), | |
('size', float), | |
('growth', float), | |
('color', float, (4,))]) | |
# 随机初始化雨滴的位置和生长速率 | |
rain_drops['position'] = np.random.uniform(0, 1, (n_drops, 2)) | |
rain_drops['growth'] = np.random.uniform(50, 200, n_drops) | |
# 创建散点图对象,用于在动画中更新雨滴的状态 | |
scat = ax.scatter(rain_drops['position'][:, 0], rain_drops['position'][:, 1], | |
s=rain_drops['size'], lw=0.5, edgecolors=rain_drops['color'], | |
facecolors='none') | |
def update(frame_number): | |
# 获取一个索引,用于重新生成最旧的雨滴 | |
current_index = frame_number % n_drops | |
# 随着时间的推移,使所有雨滴的颜色更加透明 | |
rain_drops['color'][:, 3] -= 1.0 / len(rain_drops) | |
rain_drops['color'][:, 3] = np.clip(rain_drops['color'][:, 3], 0, 1) | |
# 所有雨滴变大 | |
rain_drops['size'] += rain_drops['growth'] | |
# 为最旧的雨滴选择一个新的位置,重置其大小、颜色和生长速率 | |
rain_drops['position'][current_index] = np.random.uniform(0, 1, 2) | |
rain_drops['size'][current_index] = 5 | |
rain_drops['color'][current_index] = (0, 0, 0, 1) | |
rain_drops['growth'][current_index] = np.random.uniform(50, 200) | |
# 使用新的颜色、大小和位置更新散点图对象 | |
scat.set_edgecolors(rain_drops['color']) | |
scat.set_sizes(rain_drops['size']) | |
scat.set_offsets(rain_drops['position']) | |
# 创建动画,并将update函数作为动画的回调函数 | |
animation = FuncAnimation(fig, update, interval=10, save_count=100) | |
plt.show() |
示例-跨子图动画
import matplotlib.pyplot as plt | |
import numpy as np | |
import matplotlib.animation as animation | |
from matplotlib.patches import ConnectionPatch | |
# 创建一个包含左右两个子图的图形对象 | |
fig, (axl, axr) = plt.subplots( | |
ncols=2, # 指定一行中子图的列数为2,即创建两个子图 | |
sharey=True, # 共享y轴刻度 | |
figsize=(6, 2), | |
# width_ratios=[1, 3]指定第二个子图的宽度为第一个子图的三倍 | |
# wspace=0 设置子图之间的水平间距为0 | |
gridspec_kw=dict(width_ratios=[1, 3], wspace=0), | |
) | |
# 设置左侧子图纵横比为1,即使得它的宽度和高度相等 | |
axl.set_aspect(1) | |
# 设置右侧子图纵横比为1/3,即高度是宽度的三分之一 | |
axr.set_box_aspect(1 / 3) | |
# 右子图不显示y轴刻度 | |
axr.yaxis.set_visible(False) | |
# 设置右子图x轴刻度以及对应的标签 | |
axr.xaxis.set_ticks([0, np.pi, 2 * np.pi], ["0", r"$\pi$", r"$2\pi$"]) | |
# 在左子图上绘制圆 | |
x = np.linspace(0, 2 * np.pi, 50) | |
axl.plot(np.cos(x), np.sin(x), "k", lw=0.3) | |
# 在左子图上绘制初始点 | |
point, = axl.plot(0, 0, "o") | |
# 在右子图上绘制完整的正弦曲线,以设置视图限制 | |
sine, = axr.plot(x, np.sin(x)) | |
# 绘制连接两个图表的连线 | |
con = ConnectionPatch( | |
(1, 0), # 连接线的起始点坐标 | |
(0, 0), # 连接线的终点坐标 | |
"data", | |
"data", | |
axesA=axl, # 指定连接线的起始点所在的坐标轴 | |
axesB=axr, # 指定连接线的终点所在的坐标轴 | |
color="red", | |
ls="dotted", # 连接线类型 | |
) | |
fig.add_artist(con) | |
# 定义动画函数 | |
def animate(i): | |
x = np.linspace(0, i, int(i * 25 / np.pi)) | |
sine.set_data(x, np.sin(x)) | |
x, y = np.cos(i), np.sin(i) | |
point.set_data([x], [y]) | |
con.xy1 = x, y | |
con.xy2 = i, y | |
return point, sine, con | |
# 创建动画对象 | |
ani = animation.FuncAnimation( | |
fig, | |
animate, | |
interval=50, | |
blit=False, # 不使用blitting技术,这里Figure artists不支持blitting | |
frames=x, | |
repeat_delay=100, # 动画重复播放延迟100毫秒 | |
) | |
# 展示动画 | |
plt.show() |
示例-动态示波器
import matplotlib.pyplot as plt | |
import numpy as np | |
import matplotlib.animation as animation | |
from matplotlib.lines import Line2D | |
# 创建一个 Scope 类用于绘制动态图形 | |
class Scope: | |
def __init__(self, ax, maxt=2, dt=0.02): | |
""" | |
:param ax: Matplotlib 的坐标轴对象 | |
:param maxt: 时间的最大值,默认为2 | |
:param dt: 时间步长,默认为0.02 | |
""" | |
self.ax = ax | |
self.dt = dt | |
self.maxt = maxt | |
self.tdata = [0] # 时间数据的列表 | |
self.ydata = [0] # y轴数据的列表 | |
self.line = Line2D(self.tdata, self.ydata) # 创建一条线对象 | |
self.ax.add_line(self.line) # 将线对象添加到坐标轴上 | |
self.ax.set_ylim(-.1, 1.1) # 设置y轴范围 | |
self.ax.set_xlim(0, self.maxt) # 设置x轴范围 | |
def update(self, y): | |
""" | |
更新图形数据 | |
:param y: 新的y轴数据 | |
:return: 更新后的线对象 | |
""" | |
lastt = self.tdata[-1] | |
if lastt >= self.tdata[0] + self.maxt: # 如果当前时间超过了最大时间,重新设置数组 | |
self.tdata = [self.tdata[-1]] | |
self.ydata = [self.ydata[-1]] | |
self.ax.set_xlim(self.tdata[0], self.tdata[0] + self.maxt) | |
self.ax.figure.canvas.draw() | |
# 进行时间的计算 | |
t = self.tdata[0] + len(self.tdata) * self.dt | |
self.tdata.append(t) | |
self.ydata.append(y) | |
self.line.set_data(self.tdata, self.ydata) | |
return self.line, | |
def emitter(p=0.1): | |
"""以概率p(范围为[0, 1))返回一个随机值,否则返回0""" | |
while True: | |
v = np.random.rand() | |
if v > p: | |
yield 0. | |
else: | |
yield np.random.rand() | |
np.random.seed(0) | |
fig, ax = plt.subplots() # 创建一个图形窗口和一对坐标轴 | |
scope = Scope(ax) # 创建一个Scope对象,用于绘制动态图 | |
# 使用scope的类函数update作为更新函数 | |
ani = animation.FuncAnimation(fig, scope.update, emitter, interval=50, blit=True, save_count=100) | |
plt.show() |
示例-世界主要城市的人口数量动态展示
本示例代码和数据来自于: how-to-create-animations-in-python。这段代码支持展示自1500年到2020年期间人口数排名靠前的城市的变化趋势。该示例只是介绍简单的动态条形图绘制,更加精美的条形图绘制可使用:bar_chart_race或pandas_alive。
import pandas as pd | |
import matplotlib.pyplot as plt | |
import matplotlib.ticker as ticker | |
from matplotlib.animation import FuncAnimation | |
import matplotlib.patches as mpatches | |
# 定义一个函数,用于生成颜色列表 | |
def generate_colors(string_list): | |
num_colors = len(string_list) | |
# 使用tab10调色板,可以根据需要选择不同的调色板 | |
colormap = plt.cm.get_cmap('tab10', num_colors) | |
colors = [] | |
for i in range(num_colors): | |
color = colormap(i) | |
colors.append(color) | |
return colors | |
# 读取CSV文件,并选择所需的列 | |
# 数据地址:https://media.geeksforgeeks.org/wp-content/cdn-uploads/20210901121516/city_populations.csv | |
df = pd.read_csv('city_populations.csv', usecols=[ | |
'name', 'group', 'year', 'value']) | |
# 将年份列转换为整数型 | |
df['year'] = df['year'].astype(int) | |
# 将人口数量列转换为浮点型 | |
df['value'] = df['value'].astype(float) | |
# 获取城市分组列表 | |
group = list(set(df.group)) | |
# 生成城市分组对应的颜色字典 | |
group_clolor = dict(zip(group, generate_colors(group))) | |
# 创建城市名称与分组的字典 | |
group_name = df.set_index('name')['group'].to_dict() | |
# 定义绘制柱状图的函数 | |
def draw_barchart(year): | |
# 根据年份筛选数据,并按人口数量进行降序排序,取出最大范围的数据 | |
df_year = df[df['year'].eq(year)].sort_values( | |
by='value', ascending=True).tail(max_range) | |
ax.clear() | |
# 绘制水平柱状图,并设置颜色 | |
ax.barh(df_year['name'], df_year['value'], color=[ | |
group_clolor[group_name[x]] for x in df_year['name']]) | |
# 在柱状图上方添加文字标签 | |
dx = df_year['value'].max() / 200 | |
for i, (value, name) in enumerate(zip(df_year['value'], df_year['name'])): | |
# 城市名 | |
ax.text(value-dx, i, name, | |
size=12, weight=600, | |
ha='right', va='bottom') | |
ax.text(value-dx, i-0.25, group_name[name], | |
size=10, color='#333333', | |
ha='right', va='baseline') | |
# 地区名 | |
ax.text(value+dx, i, f'{value:,.0f}', | |
size=12, ha='left', va='center') | |
# 设置其他样式 | |
ax.text(1, 0.2, year, transform=ax.transAxes, | |
color='#777777', size=46, ha='right', | |
weight=800) | |
ax.text(0, 1.06, 'Population (thousands)', | |
transform=ax.transAxes, size=12, | |
color='#777777') | |
# 添加图例 | |
handles = [] | |
for name, color in group_clolor.items(): | |
patch = mpatches.Patch(color=color, label=name) | |
handles.append(patch) | |
ax.legend(handles=handles, fontsize=12, loc='center', bbox_to_anchor=( | |
0.5, -0.03), ncol=len(group_clolor), frameon=False) | |
# x轴的主要刻度格式化,不保留小数 | |
ax.xaxis.set_major_formatter(ticker.StrMethodFormatter('{x:,.0f}')) | |
# 将x轴的刻度位置设置在图的顶部 | |
ax.xaxis.set_ticks_position('top') | |
# 设置x轴的刻度颜色为灰色(#777777),字体大小为16 | |
ax.tick_params(axis='x', colors='#777777', labelsize=16) | |
# 清除y轴的刻度标签 | |
ax.set_yticks([]) | |
# 在x轴和y轴上设置0.01的边距 | |
ax.margins(0, 0.01) | |
# 在x轴上绘制主要网格线,线条样式为实线 | |
ax.grid(which='major', axis='x', linestyle='-') | |
# 设置网格线绘制在图像下方 | |
ax.set_axisbelow(True) | |
# 添加绘图信息 | |
ax.text(0, 1.10, f'The {max_range} most populous cities in the world from {start_year} to {end_year}', | |
transform=ax.transAxes, size=24, weight=600, ha='left') | |
ax.text(1, 0, 'Produced by luohenyueji', | |
transform=ax.transAxes, ha='right', color='#777777', | |
bbox=dict(facecolor='white', alpha=0.8, edgecolor='white')) | |
plt.box(False) | |
# 创建绘图所需的figure和axes | |
fig, ax = plt.subplots(figsize=(12, 8)) | |
start_year = 2000 | |
end_year = 2020 | |
# 设置最多显示城市数量 | |
max_range = 15 | |
# 获取数据中的最小年份和最大年份,并进行校验 | |
min_year, max_year = min(set(df.year)), max(set(df.year)) | |
assert min_year <= start_year, f"end_year cannot be lower than {min_year}" | |
assert end_year <= max_year, f"end_year cannot be higher than {max_year}" | |
# 创建动画对象,调用draw_barchart函数进行绘制 | |
ani = FuncAnimation(fig, draw_barchart, frames=range( | |
start_year, end_year+1), repeat_delay=1000, interval=200) | |
fig.subplots_adjust(left=0.04, right=0.94, bottom=0.05) | |
# 显示图形 | |
plt.show() |
结果如下:
1.2 ArtistAnimation类
ArtistAnimation构造函数的参数含义如下:
fig
:要绘制动画的Figure对象。artists
:包含了一系列绘图对象的列表,这些绘图对象将被作为动画的帧。interval
:每一帧之间的时间间隔,以毫秒为单位,默认为200。repeat
:控制动画是否重复播放,默认为True。repeat_delay
:重复动画之间的延迟时间(以毫秒为单位),默认为0。blit
:指定是否使用blitting技术来进行绘制优化,默认为False。
示例-ArtistAnimation简单使用
import matplotlib.pyplot as plt | |
import numpy as np | |
import matplotlib.animation as animation | |
fig, ax = plt.subplots() | |
# 定义函数 f(x, y),返回 np.sin(x) + np.cos(y) | |
def f(x, y): | |
return np.sin(x) + np.cos(y) | |
# 生成 x 和 y 的取值范围 | |
x = np.linspace(0, 2 * np.pi, 120) | |
y = np.linspace(0, 2 * np.pi, 100).reshape(-1, 1) | |
# ims 是一个列表的列表,每一行是当前帧要绘制的艺术品列表; | |
# 在这里我们只在每一帧动画中绘制一个艺术家,即图像 | |
ims = [] | |
# 循环生成动画的每一帧,并存入一个列表 | |
for i in range(60): | |
# 更新 x 和 y 的取值 | |
x += np.pi / 15 | |
y += np.pi / 30 | |
# 调用函数 f(x, y),并绘制其返回的图像 | |
im = ax.imshow(f(x, y), animated=True) | |
if i == 0: | |
# 首先显示一个初始的图像 | |
ax.imshow(f(x, y)) | |
# 将当前帧添加到ims中 | |
ims.append([im]) | |
# 基于ims中的绘图对象绘制动图 | |
ani = animation.ArtistAnimation(fig, ims, interval=50, blit=True, | |
repeat_delay=1000) | |
# 显示动画 | |
plt.show() |
示例-创建动态柱状图
import matplotlib.pyplot as plt | |
import numpy as np | |
import matplotlib.animation as animation | |
fig, ax = plt.subplots() | |
rng = np.random.default_rng(0) | |
# # 创建一个包含5个元素的数组,表示数据集 | |
data = np.array([20, 20, 20, 20,20]) | |
# 创建一个包含5个字符串的列表,表示数据集的标签 | |
x = ["A", "B", "C", "D","E"] | |
# 创建一个空列表,用于存储图形对象 | |
artists = [] | |
# 创建一个包含5个颜色值的列表,用于绘制图形 | |
colors = ['tab:blue', 'tab:red', 'tab:green', 'tab:purple', 'tab:orange'] | |
for i in range(20): | |
# 随机生成一个与data形状相同的数组,并将其加到data中 | |
data += rng.integers(low=0, high=10, size=data.shape) | |
# 创建一个水平条形图,并设置颜色 | |
container = ax.barh(x, data, color=colors) | |
# 设置x轴范围 | |
ax.set_xlim(0,150) | |
# 将创建的图形对象添加到列表中 | |
artists.append(container) | |
# 创建一个ArtistAnimation对象,指定图形窗口和图形对象列表以及动画间隔时间 | |
ani = animation.ArtistAnimation(fig=fig, artists=artists, interval=200) | |
plt.show() |
1.3 动画保存
Matplotlib通过plot方法创建和显示动画。为了保存动画为动图或视频,Animation类提供了save函数。save函数的常见参数如下:
filename
:保存文件的路径和名称。writer
:指定要使用的写入器(Writer)。如果未指定,则默认使用ffmpeg写入器。fps
:设置帧速率(每秒显示多少帧),默认值为None,表示使用Animation对象中的interval属性作为帧速率。dpi
:设置输出图像的分辨率,默认值为None,表示使用系统默认值。codec
:指定视频编解码器,仅当writer为ffmpeg_writer时有效。bitrate
:设置比特率,仅当writer为ffmpeg_writer时有效。extra_args
:用于传递给写入器的额外参数。metadata
:包含文件元数据的字典。extra_anim
:与主要动画同时播放的其他动画。savefig_kwargs
:传递给savefig()的关键字参数。progress_callback
:用于在保存过程中更新进度的回调函数。
writer写入器可以指定使用各种多媒体写入程序(例如:Pillow、ffpmeg、imagemagik)保存到本地,如下所示:
Writer | Supported Formats |
---|---|
~matplotlib.animation.PillowWriter | .gif, .apng, .webp |
~matplotlib.animation.HTMLWriter | .htm, .html, .png |
~matplotlib.animation.FFMpegWriter | All formats supported by ffmpeg: ffmpeg -formats |
~matplotlib.animation.ImageMagickWriter | All formats supported by imagemagick: magick -list format |
保存动图和视频的代码如下:
# 动图 | |
ani.save(filename="pillow_example.gif", writer="pillow") | |
ani.save(filename="pillow_example.apng", writer="pillow") | |
# 视频,需要安装ffmpeg | |
ani.save(filename="ffmpeg_example.mkv", writer="ffmpeg") | |
ani.save(filename="ffmpeg_example.mp4", writer="ffmpeg") | |
ani.save(filename="ffmpeg_example.mjpeg", writer="ffmpeg") |
需要注意的是动图构建对象时所设置的参数不会影响save函数,如下所示,在FuncAnimation中设置repeat=False,即动图只播放一次。但是保存的gif文件却循环播放。这是因为save函数调用了其他第三库的动图或者视频保持函数,需要重新设置参数。
import numpy as np | |
import matplotlib.pyplot as plt | |
from matplotlib.animation import FuncAnimation | |
# 创建画布和坐标轴 | |
fig, ax = plt.subplots() | |
xdata, ydata = [], [] | |
ln, = plt.plot([], [], 'r-') | |
def init(): | |
ax.set_xlim(0, 2*np.pi) | |
ax.set_ylim(-1, 1) | |
return ln, | |
def update(frame): | |
x = np.linspace(0, 2*np.pi, 100) | |
y = np.sin(x + frame/10) | |
ln.set_data(x, y) | |
return ln, | |
# 创建动画对象 | |
ani = FuncAnimation(fig, update, frames=100, interval=100, | |
init_func=init, blit=True, repeat=False) | |
ani.save(filename="pillow_example.gif", writer=writer, dpi=150) |
要解决保存动画问题,需要自定义动画保存类,如下所示:
import numpy as np | |
import matplotlib.pyplot as plt | |
from matplotlib.animation import FuncAnimation | |
from matplotlib import animation | |
# 创建画布和坐标轴 | |
fig, ax = plt.subplots() | |
xdata, ydata = [], [] | |
ln, = plt.plot([], [], 'r-') | |
def init(): | |
ax.set_xlim(0, 2*np.pi) | |
ax.set_ylim(-1, 1) | |
return ln, | |
def update(frame): | |
x = np.linspace(0, 2*np.pi, 100) | |
y = np.sin(x + frame/10) | |
ln.set_data(x, y) | |
return ln, | |
# 创建动画对象 | |
ani = FuncAnimation(fig, update, frames=100, interval=100, | |
init_func=init, blit=True, repeat=False) | |
# 创建自定义的动画写入类 | |
class SubPillowWriter(animation.PillowWriter): | |
def __init__(self, loop=1, **kwargs): | |
super().__init__(**kwargs) | |
# 将loop设置为0,表示无限循环播放;如果设置为一个大于0的数值,表示循环播放指定次数 | |
self.loop = loop | |
# 定义播放结束时,保存图片的代码 | |
def finish(self): | |
# 调用了pillow包 | |
self._frames[0].save(self.outfile, save_all=True, append_images=self._frames[1:], duration=int( | |
1000 / self.fps), loop=self.loop) | |
# 创建动画写入对象 | |
# fps=15:每秒帧数,表示动画的播放速度为每秒 15 帧。 | |
# metadata=dict(artist='luohenyueji'):元数据信息,包括艺术家信息,将被添加到生成的GIF文件中。 | |
writer = SubPillowWriter(fps=15, metadata=dict(artist='luohenyueji')) | |
ani.save(filename="pillow_example.gif", writer=writer, dpi=150) |
2 基于Pillow库生成动图
使用Pillow库生成动图非常简单。首先,准备一个包含一系列图像帧的列表。这些图像帧可以是连续的图片,每张图片表示动画的一个时间点。接下来,使用Pillow库中的save()方法将这些图像帧保存为一个gif文件。在保存动图时,还可以设置一些参数来控制动画效果。参考以下示例,可获取具体的使用说明。
示例-滑动动图
该示例展示了一种图像滑动展示的动画效果,即通过滑动渐变的方式逐步将起始黑白图片转变为目标彩色图片。所示起始图片和目标图片如下所示:
动画结果如下所示:
本示例所提供代码主要可调参数介绍如下:
-
span
(int): 分割步长,默认为100。此参数用于控制图片合并过程中的分割步长,即每次移动的距离。 -
save
(bool): 是否保存中间帧图像,默认为False。如果设置为True,则会将生成的每一帧图像保存到指定的文件夹中。 -
orient
(str): 合并方向,默认水平。可选值为'horizontal'(水平方向)或'vertical'(垂直方向)。用于控制图像的合并方向。 -
loop
(int): 循环次数,默认为0(无限循环)。设置为正整数时,动画会循环播放指定次数;设置为0时,动画会无限循环播放。 -
duration
(int): 帧持续时间(毫秒),默认为100。用于设置每一帧图像在动画中的显示时间。 -
repeat_delay
(int): 循环之间的延迟时间(毫秒),默认为500。用于设置每次循环之间的延迟时间。 -
save_name
(str): 保存动画的文件名,默认为"output"。用于设置生成的动画文件的名称。
以下是代码实现的示例。该代码首先读取起始图片和目标图片,然后指定分割位置以设置图片两侧的效果。最后,通过调整分割位置来实现滑动渐变效果。
from PIL import Image, ImageDraw | |
import os | |
def merge_image(in_img, out_img, pos, orient="horizontal"): | |
""" | |
合并图像的函数 | |
参数: | |
in_img (PIL.Image): 输入图像 | |
out_img (PIL.Image): 输出图像 | |
pos (int): 分割位置 | |
orient (str): 图像合并方向,默认水平horizontal,可选垂直vertical | |
返回: | |
result_image (PIL.Image): 合并后的图像 | |
""" | |
if orient == "horizontal": | |
# 将图像分为左右两部分 | |
left_image = out_img.crop((0, 0, pos, out_img.size[1])) | |
right_image = in_img.crop((pos, 0, in_img.size[0], in_img.size[1])) | |
# 合并左右两部分图像 | |
result_image = Image.new( | |
'RGB', (left_image.size[0] + right_image.size[0], left_image.size[1])) | |
result_image.paste(left_image, (0, 0)) | |
result_image.paste(right_image, (left_image.size[0], 0)) | |
# 添加滑动线条 | |
draw = ImageDraw.Draw(result_image) | |
draw.line([(left_image.size[0], 0), (left_image.size[0], | |
left_image.size[1])], fill=(0, 255, 255), width=3) | |
elif orient == 'vertical': | |
# 将图像分为上下两部分 | |
top_image = out_img.crop((0, 0, out_img.size[0], pos)) | |
bottom_image = in_img.crop((0, pos, in_img.size[0], in_img.size[1])) | |
# 合并上下两部分图像 | |
result_image = Image.new( | |
'RGB', (top_image.size[0], top_image.size[1] + bottom_image.size[1])) | |
result_image.paste(top_image, (0, 0)) | |
result_image.paste(bottom_image, (0, top_image.size[1])) | |
# 添加滑动线条 | |
draw = ImageDraw.Draw(result_image) | |
draw.line([(0, top_image.size[1]), (top_image.size[0], | |
top_image.size[1])], fill=(0, 255, 255), width=3) | |
return result_image | |
def main(img_in_path, img_out_path, span=100, save=False, orient='horizontal', loop=0, duration=100, repeat_delay=500, save_name="output"): | |
""" | |
主函数 | |
参数: | |
img_in_path (str): 起始图片路径 | |
img_out_path (str): 目标图片路径 | |
span (int): 分割步长,默认为100 | |
save (bool): 是否保存中间帧图像,默认为False | |
orient (str): 合并方向,默认水平 | |
loop (int): 循环次数,默认为0(无限循环) | |
duration (int): 帧持续时间(毫秒),默认为100 | |
repeat_delay (int): 循环之间的延迟时间(毫秒),默认为500 | |
save_name (str): 保存动画的文件名,默认为"output" | |
""" | |
# 读取原始图像 | |
img_in = Image.open(img_in_path).convert("RGB") | |
img_out = Image.open(img_out_path).convert("RGB") | |
assert img_in.size == img_out.size, "Unequal size of two input images" | |
if save: | |
output_dir = 'output' | |
os.makedirs(output_dir, exist_ok=True) | |
frames = [] | |
frames.append(img_in) | |
span_end = img_in.size[0] if orient == 'horizontal' else img_in.size[1] | |
# 逐张生成gif图片每一帧 | |
for pos in range(span, span_end, span): | |
print(pos) | |
result_image = merge_image(img_in, img_out, pos, orient) | |
if save: | |
result_image.save(f"output/{pos:04}.jpg") | |
frames.append(result_image) | |
if save: | |
img_in.save("output/0000.jpg") | |
img_out.save(f"output/{img_in.size[0]:04}.jpg") | |
# 添加过渡效果 | |
durations = [duration]*len(frames) | |
durations.append(repeat_delay) | |
frames.append(img_out) | |
# 生成动图 | |
# frames[0].save:表示将frames列表中的第一张图片作为输出GIF动画的第一帧 | |
# '{save_name}.gif':表示将输出的GIF动画保存在当前目录下并命名为{save_name}.gif | |
# format='GIF':表示输出的文件格式为GIF格式 | |
# append_images=frames[1:]:表示将frames列表中除了第一张图片以外的剩余图片作为输出GIF动画的后续帧 | |
# save_all=True:表示将所有的帧都保存到输出的GIF动画中 | |
# duration:表示每一帧的持续时间duration,可以是数值也可以是列表。如果是列表则单独表示每一帧的时间 | |
# loop=0:表示循环播放次数为0,即无限循环播放 | |
# optimize=True:表示优化图片生成 | |
frames[0].save(f'{save_name}.gif', format='GIF', append_images=frames[1:], | |
save_all=True, duration=durations, loop=loop, optimize=True) | |
if __name__ == "__main__": | |
# 起始图片路径 | |
img_in_path = 'in.jpg' | |
# 目标图片路径 | |
img_out_path = 'out.jpg' | |
# 调用 main 函数,并传入相应的参数 | |
main( | |
img_in_path, # 起始图片路径 | |
img_out_path, # 目标图片路径 | |
save=True, # 是否保存中间结果 | |
span=150, # 分割步长,默认为 150 | |
orient='horizontal', # 合并方向,默认为水平(可选值为 'horizontal' 或 'vertical') | |
duration=500, # 帧持续时间(毫秒),默认为500 | |
save_name="output", # 保存动画的文件名,默认为 "output" | |
repeat_delay=2000 # 循环之间的延迟时间(毫秒)默认为 500 | |
) |
上述代码演示了一种直接生成动图的方法。此外,还可以通过读取磁盘中的图片集合来生成动图。以下是示例代码,用于读取之前保存的中间图片并生成动图:
from PIL import Image | |
import os | |
# 图片文件夹路径 | |
image_folder = 'output' | |
# 保存的动图路径及文件名 | |
animated_gif_path = 'output.gif' | |
# 获取图片文件列表 | |
image_files = [f for f in os.listdir(image_folder) if f.endswith('.jpg') or f.endswith('.png')] | |
image_files.sort() | |
# 创建图片帧列表 | |
frames = [] | |
for file_name in image_files: | |
image_path = os.path.join(image_folder, file_name) | |
img = Image.open(image_path) | |
frames.append(img) | |
# 保存为动图 | |
frames[0].save(animated_gif_path, format='GIF', append_images=frames[1:], save_all=True, duration=200, loop=0) |