我正在嘗試創建一個函式，該函式將兩個不同子陣列的相同索引的元素系結到一個 numpy 陣列中。例如，如果輸入是 ，input = [[1,2,3],[4,5,6]]我想要它output =[(1,4),(2,5),(3,6)]或類似的東西。子陣列的數量可以變化（輸入是 *輸入）我要么在最后將新系結的元素相加。例如，[1 4, 2 5, 3 6]。我知道一種解決方法，但我想知道是否有一些內置函式可以做到這一點。
我試過的 -numpy.add()函式需要兩個 numpy 陣列所以這行不通
- for x,y in zip (*input[i] for i in len(lists)) 給出引數不足的錯誤
uj5u.com熱心網友回復：
如果最后你對元素的總和感興趣，我會簡單地做：
import numpy as np
input = np.array([[1,2,3],[4,5,6]])
np.sum(input, axis=0)
輸出 = 陣列([5, 7, 9])
如果你確實想在某個時候擁有中間陣列，你只需要一個轉置：
input.T
輸出 = 陣列([[1, 4], [2, 5], [3, 6]])
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/371782.html
上一篇：Python逐元素比較