numpy:将几列乘以一列

利威尔式

我想用一个两两列乘以一列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] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章