Untitled
unknown
plain_text
2 years ago
1.5 kB
19
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