我想用一个两两列乘以一列pairweise来乘以一个数组:
import numpy as np
a=np.array([[1,2],[1,2]])
b=np.array([1,0])
我想要输出:
c=[[1,2][0,0]]
那怎么可能?
最好的祝福!
您可以使用以下方法numpy.newaxis
:
import numpy as np
a = np.array([[1,2],[1,2]])
b = np.array([1,0])
c = a * b[:, np.newaxis]
# [[1 2]
# [0 0]]
或者您可以使用:
c = a * b[:, None]
这是因为newaxis
是的别名None
。
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句