我们说我有:
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])
这一点非常重要,尽快运行。
答案 0 :(得分:3)
使用这些输入(x
和y
),您可以告诉numpy向输出添加新轴,然后您将能够获得结果矩阵,其中每对{{1评估:
x, y
在这种情况下给予:
numpy.cos(x[:, numpy.newaxis])*numpy.cos(y)