How can I generate a set of points evenly distributed along the perimeter of an ellipse?

后端 未结 8 730
庸人自扰
庸人自扰 2020-12-15 22:28

If I want to generate a bunch of points distributed uniformly around a circle, I can do this (python):

r = 5  #rad         


        
8条回答
  •  长情又很酷
    2020-12-15 22:44

    (UPDATED: to reflect new packaging).

    An efficient solution of this problem for Python can be found in the numeric branch FlyingCircus-Numeric of FlyingCircus Python package.

    Disclaimer: I am the main author of them.

    Briefly, the (simplified) code looks (where a is the minor axis, and b is the major axis):

    import numpy as np
    import scipy as sp
    import scipy.optimize
    
    def angles_in_ellipse(
            num,
            a,
            b):
        assert(num > 0)
        assert(a < b)
        angles = 2 * np.pi * np.arange(num) / num
        if a != b:
            e = (1.0 - a ** 2.0 / b ** 2.0) ** 0.5
            tot_size = sp.special.ellipeinc(2.0 * np.pi, e)
            arc_size = tot_size / num
            arcs = np.arange(num) * arc_size
            res = sp.optimize.root(
                lambda x: (sp.special.ellipeinc(x, e) - arcs), angles)
            angles = res.x 
        return angles
    

    It makes use of scipy.special.ellipeinc() which provides the numerical integral along the perimeter of the ellipse, and scipy.optimize.root() for solving the equal-arcs length equation for the angles.

    To test that it is actually working:

    a = 10
    b = 20
    n = 16
    
    phi = angles_in_ellipse(n, a, b)
    print(np.round(np.rad2deg(phi), 2))
    # [  0.    16.4   34.12  55.68  90.   124.32 145.88 163.6  180.   196.4 214.12 235.68 270.   304.32 325.88 343.6 ]
    
    e = (1.0 - a ** 2.0 / b ** 2.0) ** 0.5
    arcs = sp.special.ellipeinc(phi, e)
    print(np.round(np.diff(arcs), 4))
    # [0.2829 0.2829 0.2829 0.2829 0.2829 0.2829 0.2829 0.2829 0.2829 0.2829 0.2829 0.2829 0.2829 0.2829 0.2829]
    
    # plotting
    import matplotlib.pyplot as plt
    fig = plt.figure()
    ax = fig.gca()
    ax.axes.set_aspect('equal')
    ax.scatter(b * np.sin(phi), a * np.cos(phi))
    plt.show()
    

提交回复
热议问题