将矩阵转换为特定格式的原始字节

Question

我有以下矩阵

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

，并希望将矩阵存储为语法i j 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比这更深的话，则将相互之间的理解嵌套在一起。