From c52542cff7d93da71524043d0d01791a4675900a Mon Sep 17 00:00:00 2001 From: Tom Weber Date: Sat, 12 Nov 2022 22:35:53 +0100 Subject: [PATCH] use bernoulli distribution for random start --- main.py | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/main.py b/main.py index 8418956..5951bae 100644 --- a/main.py +++ b/main.py @@ -1,6 +1,5 @@ #!/usr/bin/env python -import random import copy import matplotlib.pyplot as plt import numpy as np @@ -31,13 +30,8 @@ class State(object): ) def randomBoard(self): - for idx, i in enumerate(self.board): - for jdx, j in enumerate(i): - if random.randint(1, 2) == 1: - self.board[idx, jdx] = 1 - else: - self.board[idx, jdx] = 0 - + self.board = np.random.binomial(1, 0.5, (self.width, self.depth)) + return self.board def terminal_render(self): print("-" * (self.width + 2)) for i in self.board: