numpy 合并给定索引中的两个数组--Python

vh0rcniy  于 2022-11-10  发布在  Python
关注(0)|答案(1)|浏览(152)

我想知道是否有一种简单的方法可以将新的元素附加到数组中,但不是在数组的结尾
假设我有一个向量

a = np.array([1,2,3,4,5,6])

我想追加一个新的向量

b = np.array([1,1,1,1])

到a,从元素3开始,因此新数组将是

c = np.array([1,2,3,5,6,7,1])

也就是说,数组a的最后3个元素是由a+b产生的,而新元素只属于C
有什么主意吗?
THX
我试过了,只是添加!

iyfjxgzm

iyfjxgzm1#

pad使用NumPy:

a = np.array([1,2,3,4,5,6])
b = np.array([1,1,1,1])

# or

# a = [1,2,3,4,5,6]

# b = [1,1,1,1]

n = 3
extra = len(b)-len(a)+n

c = np.pad(a, (0, extra))
c[n:] += b

产出:

array([1, 2, 3, 5, 6, 7, 1])

相关问题