计算点积的pythonic方法是什么? [英] What is the pythonic way to calculate dot product?

问题描述

我有两个列表，一个被命名为A，另一个被命名为B.A中的每个元素都是一个三元组，而B中的每个元素只是一个数字.我想将结果定义为:

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 :

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

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

我知道逻辑很简单，但是如何以pythonic方式编写?

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

谢谢！

推荐答案

import numpy
result = numpy.dot( numpy.array(A)[:,0], B)

http://docs.scipy.org/doc/numpy/reference/

如果您希望不使用numpy，请尝试

If you want to do it without numpy, try

sum( [a[i][0]*b[i] for i in range(len(b))] )

这篇关于计算点积的pythonic方法是什么?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！