將矩陣轉換為特定格式的原始字節

Question

我有以下矩陣

B = [[1,2], [3,4]]

並想將矩陣存儲為語法ij b_ij行，其中i和j是矩陣索引， b_ij是該索引位置的值。

也就是說，上面的矩陣如下所示：

0 0 1
0 1 2
1 0 3
1 1 4

無論如何，是否可以使用Python中的庫來執行此操作？ 另外，這是用於將矩陣存儲為原始字節的通用格式嗎？ 我知道很容易遍歷矩陣以這種方式存儲它，但是效率似乎很低。

Answer 1

它不是一個庫，但是您可以使用列表理解：

>>> B = [[1, 2], [3, 4]]
>>> matrix = [ [i, j, B[i][j]] for i in range(len(B)) for j in range(len(B[i])) ]
>>> print(matrix)
[[0, 0, 1], [0, 1, 2], [1, 0, 3], [1, 1, 4]]

您也可以擴展for循環以執行類似的操作，或者如果矩陣B的深度超出此范圍，則可以相互嵌套理解。