我试图在matplotlib中获得散点图的三维动画,它基于发布的二维散点图动画here和发布的三维线图here。
这些问题产生于set_data
和set_offsets
在3D中不工作,因此您应该使用set_3d_properties
来附加z信息。玩弄它通常会窒息,但下面的代码会运行。然而,透明度的增加足以使点在几帧后消失。我在这里做错什么了?我想让这些点在盒子的范围内跳跃一段时间。即使将步长调整到非常小的值也不会降低透明度。
import matplotlib.pyplot as plt
import matplotlib.animation as animation
import numpy as np
from mpl_toolkits.mplot3d import Axes3D
FLOOR = -10
CEILING = 10
class AnimatedScatter(object):
def __init__(self, numpoints=5):
self.numpoints = numpoints
self.stream = self.data_stream()
self.angle = 0
self.fig = plt.figure()
self.ax = self.fig.add_subplot(111,projection = '3d')
self.ani = animation.FuncAnimation(self.fig, self.update, interval=100,
init_func=self.setup_plot, blit=True)
def change_angle(self):
self.angle = (self.angle + 1)%360
def setup_plot(self):
x, y, z = next(self.stream)
c = ['b', 'r', 'g', 'y', 'm']
self.scat = self.ax.scatter(x, y, z,c=c, s=200, animated=True)
self.ax.set_xlim3d(FLOOR, CEILING)
self.ax.set_ylim3d(FLOOR, CEILING)
self.ax.set_zlim3d(FLOOR, CEILING)
return self.scat,
def data_stream(self):
data = np.zeros((3, self.numpoints))
xyz = data[:3, :]
while True:
xyz += 2 * (np.random.random((3, self.numpoints)) - 0.5)
yield data
def update(self, i):
data = next(self.stream)
data = np.transpose(data)
self.scat.set_offsets(data[:,:2])
#self.scat.set_3d_properties(data)
self.scat.set_3d_properties(data[:,2:],'z')
self.change_angle()
self.ax.view_init(30,self.angle)
plt.draw()
return self.scat,
def show(self):
plt.show()
if __name__ == '__main__':
a = AnimatedScatter()
a.show()
最后找到了解决方案,下面是如何更新点而不接触颜色:
这是通过设置3d属性和重新初始化颜色在内部完成的
我找到了一个更通用的解决方案: 在将数据插入集合之前,必须添加
np.ma.ravel( x_data ) ...
。但是散点图似乎不适合动画,它太慢了。
相关问题 更多 >
编程相关推荐