Numpy：将多列与一列相乘

Question

我希望将一个数组乘以一个列对称的两列：

import numpy as np

a=np.array([[1,2],[1,2]])
b=np.array([1,0])

我想要输出：

c=[[1,2][0,0]]

这怎么可能？

祝你好运！

Answer 1

你可以使用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的别名。