python - 如何以最有效的方式在数组中添加元素？

Question

我有一个数组

myarray = [[a, b], [c, d], [e, f]]

我想得到

myarray_2 = [a + c + e, b + d + f]

我怎么能在没有循环的情况下做到这一点？什么是最有效的方法，或者可能是最 Pythonic 的方法？

标签： pythonarraysfunctiondot-product

Answer 1

您可以使用zip、sum和列表理解：

myarray = [[1, 2], [3, 4], [5, 6]]

myarray_2 = [sum(t) for t in zip(*myarray)]

print(myarray_2)

输出：

[9, 12]

无论子列表的大小如何，这都会起作用。