python - 在 Python 中添加简洁的向量？

Question

我经常做 Python 列表的向量加法。

示例:我有两个这样的列表:

a = [0.0, 1.0, 2.0]
b = [3.0, 4.0, 5.0]

我现在想将 b 添加到 a 以获得结果 a = [3.0, 5.0, 7.0]。

通常我最终会这样做:

a[0] += b[0]
a[1] += b[1]
a[2] += b[2]

有没有一些高效、标准的方法来减少输入？

更新:可以假设列表的长度为 3 并且包含 float 。

Answer 1

如果您需要高效的向量算术，请尝试 Numpy .

>>> import numpy
>>> a=numpy.array([0,1,2])
>>> b=numpy.array([3,4,5])
>>> a+b
array([3, 5, 7])
>>> 

或者(感谢安德鲁·贾菲)，

>>> a += b
>>> a
array([3, 5, 7])
>>>