将两个nparray相乘python

Question

将两个numpy数组相乘的有效方法是什么？例如，给定

A = [1, 2, 3, 4]
B = [2, 3, 5, 7]

我想计算A和B之间的点积，它是

  A.B/|A||B| = (1*2 + 2*3 + .. 4*7)/sqrt(1^2 + 2^2... +4^2) * sqrt(.....)

我如何才能高效、快速地完成这项工作？

Answer 1

如果你使用的是numpy，numpy.dot会帮你完成这项工作

numpy.dot(A,B)
51

向量的最快范数是

n = math.sqrt(numpy.dot(A,A.conj()))

下面是与其他方法的比较

>>> t1=timeit.Timer("n = math.sqrt(numpy.dot(A,A.conj()))","from __main__ import A,math,numpy")
>>> t2=timeit.Timer("n = math.sqrt(sum(abs(A)**2))","from __main__ import A,math")
>>> t3=timeit.Timer("numpy.linalg.norm(A)","from __main__ import A,numpy")
>>> print "%.2f usec/pass" % (1000000 * t1.timeit(number=100000)/100000)
2.82 usec/pass
>>> print "%.2f usec/pass" % (1000000 * t2.timeit(number=100000)/100000)
13.16 usec/pass
>>> print "%.2f usec/pass" % (1000000 * t3.timeit(number=100000)/100000)
15.68 usec/pass
>>> 

Answer 2

除了numpy.dot之外，还有numpy.linalg.norm可以满足您的需求：

from numpy.linalg import norm    
from numpy import dot

dot(a,b)/(norm(a)*norm(b))

我猜您需要的是平方和的sqrt，这是norm的默认设置。这一指标称为Frobenius范数或L2范数。如果您想要一个不同的度量标准，比如Manhattan或L1规范，那么它只是一个传入的参数。