From 7b3ab5b8ff09f82ba7a267dc46ab6c2c988448f1 Mon Sep 17 00:00:00 2001 From: Marijn Doeve Date: Sat, 4 Dec 2021 10:58:09 +0100 Subject: [PATCH] better readking --- 4.py | 16 +++++++++------- 1 file changed, 9 insertions(+), 7 deletions(-) diff --git a/4.py b/4.py index 418d778..ac03b46 100644 --- a/4.py +++ b/4.py @@ -40,20 +40,22 @@ def part_two(boards: np.ndarray, number): if __name__ == "__main__": + import sys + numbers = [int(x) for x in input().split(",")] input() # Skip blank line boards = [] - try: - while True: - current_board = [] - while (line := input().strip()) != "": - current_board.append([int(x) for x in line.split()]) + current_board = [] + + for line in sys.stdin: + if line != "\n": + current_board.append([int(x) for x in line.split()]) + else: boards.append(current_board) - except EOFError: - boards.append(current_board) + current_board = [] boards = np.array(boards)