Ensure integer division

csc-training · Apr 7, 2022 · 60bc9f2 · 60bc9f2
1 parent 006c9a7
commit 60bc9f2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/numpy/game-of-life/solution/game-of-life.py b/numpy/game-of-life/solution/game-of-life.py
@@ -16,8 +16,8 @@ def initialize(size, shape='cross'):
         board = np.random.rand(size, size).round(0).astype(int)
     elif shape == 'cross':
         board = np.zeros((size, size), int)
-        board[size/2,:] = 1
-        board[:,size/2] = 1
+        board[size//2,:] = 1
+        board[:,size//2] = 1
     else:
         raise NotImplementedError('Unknown initial shape')