如何在 python 中的 numpy 数组中绑定相同索引的元素

Question

我正在尝试制作一个 function 来绑定一个 numpy 数组中两个不同子数组的相同索引的元素。 例如，如果输入是input = [[1,2,3],[4,5,6]] ，我想要它output =[(1,4),(2,5),(3,6)]或类似的东西。 子数组的数量可以变化（输入为 *输入）我要么将新绑定的元素最后相加。 例如，[1+4、2+5、3+6]。 我知道一种解决方法，但我想知道是否有一些内置函数可以做到这一点。

我试过的 - numpy.add() function 需要两个 numpy ZA3CBC3F9D06'12F2C1954DE1

• for x,y in zip (*input[i] for i in len(lists)) 给出 arguments 不足的错误

Answer 1

如果最终您对元素的总和感兴趣，我会这样做：

import numpy as np
input = np.array([[1,2,3],[4,5,6]])
np.sum(input, axis=0)

输出 = 数组（[5, 7, 9]）

如果你确实想在某个时候拥有中间数组，你只需要一个转置：

input.T

output = 数组（[[1, 4], [2, 5], [3, 6]]）