entity_results = [] captured_results = [] for trail in range(N_TRIALS): # Set up the environment env = Environment(ENV_SIZE, ENV_SIZE, N_PATCHES) entities = random.randint(MIN_ENTITIES_PER_PATCH, MAX_ENTITIES_PER_PATCH) for patch in env.children: patch.create_entities(entities) s = Searcher(x_pos=(env.length / 2.0), y_pos=(env.width / 2.0), parent=env) for i in range(MAX_MOVES): s.move() entity = s.detect() s.capture(entity) entity_results.append(entities) captured_results.append(len(s.captured)) # for k in range(len(entity_results)): # print(entity_results[k], captured_results[k]) x = np.array(entity_results) y = np.array(captured_results) slope, intercept, r_value, p_value, slope_std_error = stats.linregress(x, y) print "Slope, intercept:", slope, intercept print "R-squared:", r_value**2