乘以不同大小的numpy数组

Question

我们说我有：

x = numpy.array([1,2,3])
y = numpy.array([1,2])

我想要的是一种矢量化和有效的方式：

numpy.cos(x)*numpy.cos(y[0])
numpy.cos(x)*numpy.cos(y[1])

这一点非常重要，尽快运行。

Answer 1

使用这些输入（x和y），您可以告诉numpy向输出添加新轴，然后您将能够获得结果矩阵，其中每对{{1评估：

x, y

在这种情况下给予：

numpy.cos(x[:, numpy.newaxis])*numpy.cos(y)