python - 在 Numpy 中将向量附加到矩阵的优雅解决方案？

Question

我看到其他人发过这个帖子，但我不清楚是否有更好的解决方案。我有一个 2D NumPy 数组，我想向它追加一列。例如:

import numpy as np
A = np.array([[2., 3.],[-1., -2.]])
e = np.ones(2)
print(A)
print(e)
B = np.hstack((A,e.reshape((2,1)))) 
print(B)

做的正是我想要的。但是有没有办法避免这种笨拙的 reshape 使用？

Answer 1

如果您想避免使用 reshape，那么您必须附加一个尺寸正确的列:

e = np.ones((2, 1))
B = np.hstack((A,e))

注意对 ones 调用的修改。您现在必须使用 reshape 的原因是 numpy 不认为维度 2 的数组与维度 (2, 1) 的数组相同。第二个是二维数组，其中一个维度的大小为 1。