ma_carmunk.py 文件源码

def makeCars(self):
        index1= list(range(len(self.lanePos1)))
        index2 = list(range(len(self.lanePos2)))
        shuffle(index1)
        shuffle(index2)
        color_pro = np.random.sample(self.numCar) < 0.5
        lane_pro = np.random.sample(self.numCar) < 0.5

        car = []
        for i in xrange(0, self.numCar):
            color = 1
            if color_pro[i]:
                color = 0
            if lane_pro[i]: ## make use pop method.
                idx = index1.pop()
                lane = self.lanePos1[idx]
            else:
                idx = index2.pop()
                lane = self.lanePos2[idx]
            car.append(Car(0, lane[0], lane[1], self.space, self.width, self.height, goal = self.goal, 
                car_radius=self.car_radius, agentId=i, color=color))
        return car