Как мне взять матрицу, отформатированную так, как ввод в python3

Question

Я понимаю, что есть похожие вопросы, но ни у одного не было ответа на этот конкретный c тип форматирования. Для проблемы codeforces мне нужно взять матричный вход как такой:

0 0 0 0 0
0 0 0 0 1
0 0 0 0 0
0 0 0 0 0
0 0 0 0 0

, и я хочу сохранить его как матрицу, где m [i] [j] точно так же, как и вход выше. Например, я хочу сохранить входные данные в виде матрицы, где [4] [1] будет соответствовать 1. Как я могу принять этот входной сигнал в python3? В случае, если это полезно, я приложил проблему здесь: https://codeforces.com/problemset/problem/263/A. Единственное, что следует отметить, это то, что матрица всегда 5x5.

Answer 1

Использование:

N = 5 # Shape of the square matrix(NxN)
matrix = [list(map(int, input().split())) for _ in range(N)]

print(matrix)
print("Element at 2 row and 5 column is:", matrix[1][4])

Выход:

[[0, 0, 0, 0, 0], 
 [0, 0, 0, 0, 1], 
 [0, 0, 0, 0, 0], 
 [0, 0, 0, 0, 0], 
 [0, 0, 0, 0, 0]]

Element at 2 row and 5 column is: 1

Answer 2

Было бы намного более интуитивно понятно, если вы поменяете местами [4] [1] и обозначите матрицу [1] [4] = 1. В этом случае вы можете просто сохранить матрицу в виде вложенного списка, где каждый подсписок это строка. матрица [1] [4] будет трассироваться обратно к элементу с индексом строки 1 и индексом столбца 4