Анимация трехмерного рассеянного графика с использованием Python mplotlib через последовательные данные

Question

Я пытаюсь анимировать трехмерный график рассеяния, используя mplotlib в Python.Я могу отображать данные и перерисовывать их каждый раз, но это приводит к частоте кадров менее 1 FPS, и мне нужно увеличить до 30 FPS.Когда я запускаю свой код:

import serial
import numpy
import matplotlib.pyplot as plt #import matplotlib library
from mpl_toolkits.mplot3d import Axes3D
from drawnow import *
import matplotlib.animation
import time

ser = serial.Serial('COM7',9600,timeout=5)
ser.flushInput()
time.sleep(5)
ser.write(bytes(b's1000'))

x=list()
y=list()
z=list()

#plt.ion()
fig = plt.figure(figsize=(16,12))
ax = fig.add_subplot(111, projection="3d")
graph = ax.scatter(x,y,z, c='r',marker='o')
ax.set_xlim3d(-255, 255)
ax.set_ylim3d(-255, 255)
ax.set_zlim3d(-255, 255)

def generate():
    while True:
        try:
            ser_bytes = ser.readline()
            data = str(ser_bytes[0:len(ser_bytes)-2].decode("utf-8"))
            xyz = data.split(", ")
            dx = float(xyz[0])
            dy = float(xyz[1])
            dz = float(xyz[2].replace(";",""))
            x.append(dx);
            y.append(dy);
            z.append(dz);
            graph._offset3d(x,y,z, c='r',marker='o')

        except:
            print("Keyboard Interrupt")
            ser.close()
            break
    return graph,

ani = matplotlib.animation.FuncAnimation(fig, generate(), interval=1, blit=True)
plt.show()

я получаю следующую ошибку:

Traceback (most recent call last):
  File "C:\Users\bunti\AppData\Local\Programs\Python\Python36-32\lib\site-packages\matplotlib\cbook\__init__.py", line 388, in process
    proxy(*args, **kwargs)
  File "C:\Users\bunti\AppData\Local\Programs\Python\Python36-32\lib\site-packages\matplotlib\cbook\__init__.py", line 228, in __call__
    return mtd(*args, **kwargs)
  File "C:\Users\bunti\AppData\Local\Programs\Python\Python36-32\lib\site-packages\matplotlib\animation.py", line 1026, in _start
    self._init_draw()
  File "C:\Users\bunti\AppData\Local\Programs\Python\Python36-32\lib\site-packages\matplotlib\animation.py", line 1750, in _init_draw
    self._draw_frame(next(self.new_frame_seq()))
  File "C:\Users\bunti\AppData\Local\Programs\Python\Python36-32\lib\site-packages\matplotlib\animation.py", line 1772, in _draw_frame
    self._drawn_artists = self._func(framedata, *self._args)
TypeError: 'tuple' object is not callable
Traceback (most recent call last):
  File "C:\Users\bunti\AppData\Local\Programs\Python\Python36-32\lib\site-packages\matplotlib\cbook\__init__.py", line 388, in process
    proxy(*args, **kwargs)
  File "C:\Users\bunti\AppData\Local\Programs\Python\Python36-32\lib\site-packages\matplotlib\cbook\__init__.py", line 228, in __call__
    return mtd(*args, **kwargs)
  File "C:\Users\bunti\AppData\Local\Programs\Python\Python36-32\lib\site-packages\matplotlib\animation.py", line 1308, in _handle_resize
    self._init_draw()
  File "C:\Users\bunti\AppData\Local\Programs\Python\Python36-32\lib\site-packages\matplotlib\animation.py", line 1750, in _init_draw
    self._draw_frame(next(self.new_frame_seq()))
  File "C:\Users\bunti\AppData\Local\Programs\Python\Python36-32\lib\site-packages\matplotlib\animation.py", line 1772, in _draw_frame
    self._drawn_artists = self._func(framedata, *self._args)
TypeError: 'tuple' object is not callable

Я получаю координаты x, y, z от лидарного модуля, подключенного к Arduino, который отправляеткоординаты по последовательному к сценарию Python.

Answer 1

Возможная проблема с вашим кодом в том, что функция анимации (generate) не должна выполняться в бесконечном цикле.Кроме того, вы должны передать ссылку на эту функцию в FuncAnimate, но вместо этого вы вызываете функцию (т. Е. Вам нужно опустить скобки в FuncAnimation(..., generate, ...)

. Вторая проблема заключается в том, что вы лечитеgraph._offset3d как если бы это была функция, когда это просто кортеж списков. Вы должны назначить ему новый кортеж, вместо того, чтобы пытаться вызвать функцию (что, как я считаю, является сообщением об ошибке)намекает на).

Я упростила ваш код, и отлично работает следующее:

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.animation as animation


def update_lines(num):
    dx, dy, dz = np.random.random((3,)) * 255 * 2 - 255  # replace this line with code to get data from serial line
    text.set_text("{:d}: [{:.0f},{:.0f},{:.0f}]".format(num, dx, dy, dz))  # for debugging
    x.append(dx)
    y.append(dy)
    z.append(dz)
    graph._offsets3d = (x, y, z)
    return graph,


x = [0]
y = [0]
z = [0]

fig = plt.figure(figsize=(5, 5))
ax = fig.add_subplot(111, projection="3d")
graph = ax.scatter(x, y, z, color='orange')
text = fig.text(0, 1, "TEXT", va='top')  # for debugging

ax.set_xlim3d(-255, 255)
ax.set_ylim3d(-255, 255)
ax.set_zlim3d(-255, 255)

# Creating the Animation object
ani = animation.FuncAnimation(fig, update_lines, frames=200, interval=50, blit=False)
plt.show()