將兩個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)) 

我猜你想要的平方和，這是norm默認的開方。這個度量稱爲Frobenius範數或L2範數。如果你想要一個不同的度量標準，比如Manhattan或者L1標準，它只是一個參數。