Untitled
unknown
plain_text
a year ago
1.5 kB
5
Indexable
import numpy as np row1 = np.array(list(map(int, input().split()))) row2 = np.array(list(map(int, input().split()))) row3 = np.array(list(map(int, input().split()))) row4 = np.array(list(map(int, input().split()))) direction = int(input()) matrix = np.vstack((row1, row2, row3, row4)) def left(matrix): for i in range(3): for j in range(1, 3): matrix[i][j - 1] += matrix[i][j] matrix[i][j] = 0 matrix[i][3] += matrix[i][0] matrix[i][0] = 0 return matrix def up(matrix): for j in range(3): for i in range(1, 3): matrix[i - 1][j] += matrix[i][j] matrix[i][j] = 0 matrix[3][j] += matrix[0][j] matrix[0][j] = 0 return matrix def right(matrix): for i in range(3): for j in range(1, 3): matrix[i][j + 1] += matrix[i][j] matrix[i][j] = 0 matrix[i][0] += matrix[i][3] matrix[i][3] = 0 return matrix def down(matrix): for j in range(3): for i in range(1, 3): matrix[i + 1][j] += matrix[i][j] matrix[i][j] = 0 matrix[0][j] += matrix[3][j] matrix[3][j] = 0 return matrix if direction == 0: result = left(matrix) elif direction == 1: result = up(matrix) elif direction == 2: result = right(matrix) elif direction == 3: result = down(matrix) print(result)
Editor is loading...
Leave a Comment