env_generator.py 文件源码

def sample_spatial_poisson_process(self, rate):
        xmin, xmax = self.x_range
        ymin, ymax = self.y_range

        dx = xmax - xmin
        dy = ymax - ymin

        N = stats.poisson( rate * dx * dy ).rvs()
        x = stats.uniform.rvs(xmin, dx, ((N, 1)) )
        y = stats.uniform.rvs(ymin, dy, ((N, 1)) )

        centers = np.hstack((x,y))
        return centers