Matplotlib 散点图动画

Question

我正在尝试在 matplotlib 中绘制动画散点图。

到目前为止我写的代码：

import math
import sympy
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as ani

def convert_to_polar_coordinates(num):
    return num * np.cos(num), num * np.sin(num)


fig = plt.figure(figsize=(8, 8))
primes = sympy.primerange(0, 15000)
primes = np.array(list(primes))
x, y = convert_to_polar_coordinates(primes)
scat = plt.scatter([],[],s=1)

def init():
    scat.set_offsets([])
    return scat,

def plot_in_polar(i=int):
    data = np.hstack((x[:i,np.newaxis], y[:i, np.newaxis]))
    scat.set_offsets(data)
    return scat,

plt.axis("off")
animator = ani.FuncAnimation(fig, plot_in_polar, init_func=init, interval = 20)
plt.show()

animator.save(r'D:\primes.gif')

我在 ValueError: 'vertices' must be a 2D list or array with shape Nx2

旁边得到一个空图

我试图在 SO 上关注一些类似的问题，但我找不到我在做什么不同的事情。

Answer 1

我已将您的动画重新构造如下，如果可以，请告诉我：

import sympy
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as ani


def convert_to_polar_coordinates(num):
    return num*np.cos(num), num*np.sin(num)


lim = 5000

fig, ax = plt.subplots(figsize = (8, 8))
primes = sympy.primerange(0, lim)
primes = np.array(list(primes))
x, y = convert_to_polar_coordinates(primes)


def init():
    ax.cla()
    ax.plot([], [])


def plot_in_polar(i):
    ax.cla()
    ax.plot(x[:i], y[:i], linestyle = '', marker = 'o', markersize = 1)
    ax.set_xlim(-lim, lim)
    ax.set_ylim(-lim, lim)
    ax.axis("off")


animator = ani.FuncAnimation(fig = fig, func = plot_in_polar, interval = 20, frames = len(primes))

plt.show()

Matplotlib 散点图动画

Matplotlib scatterplot animation

python

animation

numpy

matplotlib

sympy