如何在两个向量之间插入两个向量之和？

Question

如何在它们之间插入每两个向量之和？ 示例：

[[10,10], [20,30], [1,1]] -----> [[10,10], [30,40], [20,30], [21,31], [1,1]]

我想我将首先创建一个numpy数组以获取总和，但是我不确定如何在正确的位置插入新矢量。我该在数组上执行此操作还是应该将总和转换回列表？

Answer 1

您可以计算总和，然后使用函数insert：

arr = np.array([[10,10], [20,30], [1,1]])

arr1 = arr[::2] + arr[1::2]
arr2 = np.insert(arr, np.arange(1, arr.shape[0]), arr1, axis=0)

输出：

array([[10, 10],
       [30, 40],
       [20, 30],
       [21, 31],
       [ 1,  1]])