Question

我正在尝试使用matplotlib的动画模块进行实时散点图，但我是一个新手。我的目标是每当收到我想要绘制的数据时更新绘图，这样无论何时收到数据，先前的点都会消失，而新的点会被绘制出来。

如果我用无限循环和随机生成的数据替换接收数据，我的程序就可以这样编写：

fig = plt.figure()
skyplot = fig.add_subplot(111, projection='polar')
skyplot.set_ylim(90)  # sets radius of the circle to maximum elevation
skyplot.set_theta_zero_location("N")  # sets 0(deg) to North
skyplot.set_theta_direction(-1)  # sets plot clockwise
skyplot.set_yticks(range(0, 90, 30))  # sets 3 concentric circles
skyplot.set_yticklabels(map(str, range(90, 0, -30)))  # reverse labels
plt.ion()

while(1):

    azimuths = random.sample(range(360), 8)
    elevations = random.sample(range(90), 8)
    colors = numpy.random.rand(3,1)

    sat_plot = satellite()
    ani= animation.FuncAnimation(fig, sat_plot.update, azimuths, elevations, colors)

class satellite:

    def __init__(self):
        self.azimuths = []
        self.elevations = []
        self.colors = []
        self.scatter = plt.scatter(self.azimuths, self.elevations, self.colors)

    def update(self, azimuth, elevation, colors):
        self.azimuths = azimuth
        self.elevations = elevation
        return self.scatter

现在，我收到以下错误：

> Traceback (most recent call last):
  File "./skyplot.py", line 138, in <module>
    ani= animation.FuncAnimation(fig, sat_plot.update, azimuths, elevations, colors)
  File "/usr/lib/pymodules/python2.7/matplotlib/animation.py", line 442, in __init__
    TimedAnimation.__init__(self, fig, **kwargs)
  File "/usr/lib/pymodules/python2.7/matplotlib/animation.py", line 304, in __init__
    Animation.__init__(self, fig, event_source=event_source, *args, **kwargs)
  File "/usr/lib/pymodules/python2.7/matplotlib/animation.py", line 53, in __init__
    self._init_draw()
  File "/usr/lib/pymodules/python2.7/matplotlib/animation.py", line 469, in _init_draw
    self._drawn_artists = self._init_func()
TypeError: 'list' object is not callable

任何人都可以告诉我我做错了什么，我怎么能这样做？

提前致谢

Answer 1

我认为你不需要动画。你需要一个简单的无限循环（例如while）和一个线程中的绘图更新。我可以提出这样的建议：

import threading,time
import matplotlib.pyplot as plt
import numpy as np

fig = plt.figure()
data = np.random.uniform(0, 1, (5, 3))
plt.scatter(data[:, 0], data[:,1],data[:, 2]*50)

def getDataAndUpdate():
    while True:
        """update data and redraw function"""
        new_data = np.random.uniform(0, 1, (5, 3))
        time.sleep(1)
        plt.clf()
        plt.scatter(new_data[:, 0], new_data[:, 1], new_data[:, 2] * 50)
        plt.draw()

t = threading.Thread(target=getDataAndUpdate)
t.start()
plt.show()

结果是一个带有散点图的类似动画的图形。

使用动画进行分散更新

1 个答案: