我想将矩阵的每一列添加到一个numpy数组,但numpy.broadcast
只允许将矩阵的每一行添加到一个数组。我怎样才能做到这一点?
我的想法是先转置矩阵,然后将其添加到数组中,然后转回,但这使用了两个转置。有直接执行的功能吗?
除了使用数组,您还可以使用仅包含一列的第二个矩阵:
matrix = np.matrix(np.zeros((3,3)))
array = np.matrix([[1],[2],[3]])
matrix([[1],
[2],
[3]])
matrix + array
matrix([[ 1., 1., 1.],
[ 2., 2., 2.],
[ 3., 3., 3.]])
如果您最初有一个数组,则可以像这样重塑它:
a = np.asarray([1,2,3])
matrix + np.reshape(a, (3,1))
matrix([[ 1., 1., 1.],
[ 2., 2., 2.],
[ 3., 3., 3.]])
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句