import numpy as np def uniform_sphere(size=None): """ Generate random points isotropically distributed across the unit sphere. Args: - size: int, *optional* Number of points to generate. If no size is specified, a single point is returned. source: Weisstein, Eric W. "Sphere Point Picking." Mathworld. """ theta, u = np.random.uniform(0.0, 2*np.pi, size), \ np.random.uniform(-1.0, 1.0, size) c = np.sqrt(1-u**2) points = np.empty((x.size, 3)) points[:,0] = c*np.cos(theta) points[:,1] = c*np.sin(theta) points[:,2] = u if size is None: return points[0] else: return points