如何將這樣格式的矩陣作為python3中的輸入

Question

我知道有類似的問題，但沒有人對這種特定類型的格式有答案。 對於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

如果交換 [4][1] 並表示矩陣 [1][4] = 1，會更直觀。在這種情況下，您可以將矩陣存儲為嵌套列表，其中每個子列表是一排。 matrix[1][4] 將追溯到行索引 1 和列索引 4 處的元素

Answer 2

利用：

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])

Output：

[[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