在NumPy中向矩阵添加单列数

Question

我有一个函数，它把一的列向量加到矩阵上。在

def add_ones(x):
    return np.hstack((np.ones((x.shape[0], 1)), x))

当x是一个矩阵时，它做得很好。例如：

^{pr2}$

给予

[[1. 1. 2.]
 [1. 3. 4.]]
[[1. 0. 0. 0. 0. 0.]
 [1. 0. 0. 0. 0. 0.]
 [1. 0. 0. 0. 0. 0.]
 [1. 0. 0. 0. 0. 0.]
 [1. 0. 0. 0. 0. 0.]]

但是当我传递一个向量时：

z = np.zeros(5)
add_ones(z)

它给出了以下错误：

ValueError: all the input arrays must have same number of dimensions

我想把它还回去

[1. 0. 0. 0. 0. 0.]

我该怎么办？在

Answer 1

必须向z添加另一个维度（使其成为二维5x1矩阵）：

add_ones(z[:, np.newaxis])
#array([[ 1.,  0.],
 #      [ 1.,  0.],
#       [ 1.,  0.],
#       [ 1.,  0.],
#       [ 1.,  0.]])