What is the pythonic way to calculate dot product?

python dot-product
xiao 啸 picture xiao 啸 · May 7, 2011 · Viewed 102.9k times · Source

I have two lists, one is named as A, another is named as B. Each element in A is a triple, and each element in B is just an number. I would like to calculate the result defined as :

result = A[0][0] * B[0] + A[1][0] * B[1] + ... + A[n-1][0] * B[n-1]

I know the logic is easy but how to write in pythonic way?

Thanks!

Answer

Henri Andre picture Henri Andre · Sep 23, 2016

Python 3.5 has an explicit operator @ for the dot product, so you can write

a = A @ B

instead of

a = numpy.dot(A,B)

Related questions

dot product in python

Does this Python code actually find the dot product of two vectors? import operator vector1 = (2,3,5) vector2 = (3,4,6) dotProduct = reduce( operator.add, map( operator.mul, vector1, vector2))

Read More
Dot product of two vectors in tensorflow

I was wondering if there is an easy way to calculate the dot product of two vectors (i.e. 1-d tensors) and return a scalar value in tensorflow. Given two vectors X=(x1,...,xn) and Y=(y1,...,yn), the dot …

Read More
Understanding tensordot

After I learned how to use einsum, I am now trying to understand how np.tensordot works. However, I am a little bit lost especially regarding the various possibilities for the parameter axes. To understand it, as I have never …

Read More