python - 通过添加到末尾来扩展矩阵中的每一行？

Question

我试图通过在每行末尾附加 1 来增加矩阵中的行长度。

matrix_1 = [[-4, -2, -3],[-1, -1, 1],[-2, 0, 1]]

我需要的输出是:

matrix_2 = [[-4,-2,-3,1],[-1,-1,1,1],[-2,0,1,1]]

谢谢

Answer 1

如果您正在使用此矩阵进行大量数值工作，您可能会受益于使用 numpy(基本上是 Python 线性代数库)。如果您的矩阵是 numpy 数组而不是嵌套列表，如下所示:

import numpy as np
matrix_1 = np.array([[-4, -2, -3],[-1, -1, 1],[-2, 0, 1]])

然后你可以使用 numpy 的 hstack 来扩展它功能:

column_of_ones = np.ones((matrix_1.shape[0], 1))
matrix_2 = np.hstack((matrix_1, column_of_ones))