Python以numpy生成3D球体

Question

因此，我目前制作了一个由零组成的numpy 3D数组，但是我试图创建一堆可以由numpy数组表示的不同球体。但是，我对如何真正做到这一点感到困惑。如果我有指定的中心和指定的半径，如何创建一个表示球体的numpy数组？

Answer 1

def create_sphere(cx,cy,cz, r, resolution=360):
    phi = np.linspace(0, 2*np.pi, 2*resolution)
    theta = np.linspace(0, np.pi, resolution)

    theta, phi = np.meshgrid(theta, phi)

    r_xy = r*np.sin(theta)
    x = cx + np.cos(phi) * r_xy
    y = cy + np.sin(phi) * r_xy
    z = cz + r * np.cos(theta)

    return np.stack([x,y,z])