使用Python中的matplotlib.animation制作动画3D条形图的示例

Question

我已经看到了一些使用matplotlib.animation模块的好例子，包括这个animated 3D plot example。我想知道这个动画模块是否可以与bar3d图表一起使用。

有人可以生成一个简单的例子吗？

注意：我目前正在开发一个不包含matplotlib.animation的其他解决方案（请参阅我的其他post），但这似乎是太慢......

Answer 1

这是一个2x2小条的小例子，它将随机增长和改变颜色，一次调用update_bars()：

import matplotlib.pyplot as plt
import mpl_toolkits.mplot3d.axes3d as p3
import matplotlib.animation as animation
import random

def update_bars(num, bars):
    i = random.randint(0, 3)
    dz[i] += 0.1
    bars[i] = ax.bar3d(xpos[i], ypos[i], zpos[i], dx[i], dy[i], dz[i], color=random.choice(['r', 'g', 'b']))
    return bars

fig = plt.figure()
ax = p3.Axes3D(fig)
xpos = [1, 1, 3, 3]
ypos = [1, 3, 1, 3]
zpos = [0, 0, 0, 0]
dx = [1, 1, 1, 1]
dy = [1, 1, 1, 1]
dz = [3, 2, 6, 5]

# add bars
bars = []
for i in range(4):
    bars.append(ax.bar3d(xpos[i], ypos[i], zpos[i], dx[i], dy[i], dz[i], color=random.choice(['r', 'g', 'b'])))
ax.set_title('3D bars')

line_ani = animation.FuncAnimation(fig, update_bars, 20, fargs=[bars], interval=100, blit=False)
plt.show()

输出（此处未设置动画）：