计算点积的Pythonic方法是什么？

Question

我有两个列表，一个名为 A，另一个名为 B。A 中的每个元素都是一个三元组，B 中的每个元素只是一个数字。我想计算的结果定义为：

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

我知道逻辑很简单，但如何以Pythonic方式编写？

谢谢！

Answer 1

Python 3.5 有一个用于点积的显式运算符 @，
所以你可以写

a = A @ B

而不是

a = numpy.dot(A,B)

Answer 2

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

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

如果您想要在没有 numpy 的情况下做到这一点，请尝试

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

Answer 3

我最喜欢的 Pythonic 点产品是：

sum([i*j for (i, j) in zip(list1, list2)])

因此，对于您的情况，我们可以这样做：

sum([i*j for (i, j) in zip([K[0] for K in A], B)])

Answer 4

from operator import mul

sum(map(mul, A, B))

Answer 5

使用运算符和 itertools 模块：

from operator import mul
from itertools import imap

sum(imap(mul, A, B))

Answer 6

使用more_itertools，一个实现dotproduct itertools 配方：

import more_itertools as mit


a = [1, 2, 3]
b = [7, 8, 9]

mit.dotproduct(a, b)
# 50

Answer 7

人们正在将 @ 运算符重新指定为点积运算符。这是我的代码，使用普通 python 的 zip 返回一个元组。然后使用列表理解而不是映射。

def dot_product(a_vector,b_vector):
    #a1 x b1 + a2 * b2..an*bn return scalar
    return sum([an*bn for an,bn in zip(a_vector,b_vector)])

X = [2,3,5,7,11]
Y = [13,17,19,23,29]
print(dot_product(X,Y)) #652

a=[1,2,3]
b=[4,5,6]
print(dot_product(a,b)) #prints 32= 1*4 + 2*5 + 3*6 = 
a = [1, 2, 3]
b = [7, 8, 9]
print(dot_product(a,b)) #prints 50 

Answer 8

>>> X = [2,3,5,7,11]
>>> Y = [13,17,19,23,29]
>>> dot = lambda X, Y: sum(map(lambda x, y: x * y, X, Y))
>>> dot(X, Y)
652

就是这样。

Answer 9

在 Python 3.12+ 中，您可以使用 math.sumprod。

鉴于您的情况，

A = [
    (1, 11, 111),
    (2, 22, 222),
]
B = [10, 100]

first_column = (row[0] for row in A)

您可以计算 A[0][0] * B[0] + A[1][0] * B[1]

>>> import math
>>> math.sumprod(first_column, B)
210

---

在旧版本中，最Pythonic的方法可能是使用使用生成器求和（即不构建中间列表）：

>>>  sum(row[0] * b for row, b in zip(A, B, strict=True))
210

Answer 10

对于这种事情来说，最Pythonic的方法可能是使用 numpy< /a>. ;-)

Answer 11

但是，这可能是重复的解决方案：

>>> u = [(1, 2, 3), (4, 5, 6)]
>>> v = [3, 7]

在普通 Python 中：

>>> sum([x*y for (x, *x2), y in zip(u,v)])
31

或使用 numpy （如 用户57368的回答）：

import numpy as np
>>> np.dot(np.array(u)[:,0], v)
31

Answer 12

以上所有答案都是正确的，但在我看来，计算点积的最Pythonic方法是：

>>> a=[1,2,3]
>>> b=[4,5,6]
>>> sum(map(lambda pair:pair[0]*pair[1],zip(a,b)))
32