python - 如何更新复根图

Question

我想连续绘制几个Jacobi polynomials的（复）根。这些多项式的参数和次数随大小因子 n 变化。到目前为止，我已经有了计算根并将它们绘制在复平面上的代码，但我希望每个图都覆盖前一个图，以便查看是否会发展出根的模式。

我现在编写的代码（使用numpy和sympy）产生了这些图，但它们都是独立的。

import matplotlib.pyplot as plt
import sympy
from sympy import *
import numpy
x = sympy.symbols('x')

def show_roots(n,a,b,c):
    for k in range (1,n+1):
        p = sympy.jacobi(a*k,-(a+b)*k,(b+c)*k,x)
        coeff = sympy.poly(p).coeffs()
        roots = numpy.roots(coeff)
        X = [x.real for x in roots]
        Y = [x.imag for x in roots]
        plt.scatter(X,Y, color='red')
        plt.show()
        plt.pause(0.5)

标签： pythonmatplotlibplotsympypolynomials