python - 将新条目插入一维 numpy 数组而不干扰其他条目

考虑以下内容:如果我有一个如下所示的一维数组

import numpy as np
x=np.array([1,2,4,5])

现在假设我有数字 3,并希望将它输入到数组的第三个位置(或“python 语言”中的第二个位置)而不分散其他条目,这样一旦我们在函数之后再次调用数组,我们获得:

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

由于没有更好的词,我正在寻找一种涉及数组 x 和数字 3 的“拼接方法”。

谢谢!

回答1

您可以使用 numpy.insert

numpy.insert(arr, obj, values, axis=None)

arr -> 要插入 value/values 的数组

obj -> 可以是整数、切片或 values 序列

values -> 你可以想象,你想要插入的 value/values

axis -> 它也适用于多维数组

这是您提出的已解决示例:

import numpy as np

idx = 2
val = 3

x = np.array([1,2,4,5])
x = np.insert(x,idx,val,axis=0)

这是结果

pre-insert : [1 2 4 5]
post-insert : [1 2 3 4 5]

如您所见,此方法将在 value val=3 中放入数组 x 的位置 idx=2

相似文章

随机推荐

最新文章