[英]multiplying numpy arrays of different size
假设我有:
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])
尽快运行非常重要。
使用这些输入( x
和y
),您可以告诉numpy将新轴添加到输出中,然后您将能够获得结果矩阵,其中每对x, y
被求值:
numpy.cos(x[:, numpy.newaxis])*numpy.cos(y)
在这种情况下给出:
# y[0] y[1]
array([[ 0.29192658, -0.2248451 ], # x[0]
[-0.2248451 , 0.17317819], # x[1]
[-0.53489523, 0.41198225]]) # x[2]
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.