python - Numpy 追加到矩阵

Question

我有一个函数需要在给定矩阵的开头添加一列。我已经开始工作了:

def add_ones(X):
    return np.vstack((np.ones(X.shape[0]), X.T)).T

这行得通，但如您所见，它转置了矩阵两次。我首先在不转置矩阵的情况下尝试了它，但似乎 np.ones(m) 总是产生一个我无法转置的行向量(我尝试做 np.ones(m) .T).

我的问题是，有没有更好的方法来做到这一点？我对 numpy 不是很熟悉，也不知道转置矩阵的性能下降。

Answer 1

您可以使用 np.concatenate -

np.concatenate((np.ones((X.shape[0],1)),X),axis=1)

您还可以使用 np.column_stack -

np.column_stack((np.ones((X.shape[0])),X))

最后一个 np.hstack -

np.hstack((np.ones((X.shape[0],1)),X))