gpt4 book ai didi

python - 将元素插入 numpy 数组的更好方法

转载 作者:行者123 更新时间:2023-11-30 22:07:53 26 4
gpt4 key购买 nike

我有一个 numpy 数组,并且有一个要在特定位置(不连续)插入该数组的元素列表。索引位于另一个 numpy 数组中。

target answer: [1,2,3,4,5]
original array: [1,3,5]
elements to insert: [2,4]
indices: [1,3]

numpy.insert(arr,[1,3],[2,4]) 未给出所需的结果。它给出[1,2,3,5,4]。有什么指点吗?

最佳答案

通过 np.insert 使用范围偏移索引 -

np.insert(a, add_idx - np.arange(len(add_idx)), add_val)

示例运行 -

In [20]: a
Out[20]: array([1, 3, 5])

In [21]: add_idx
Out[21]: [1, 3]

In [22]: add_val
Out[22]: [2, 4]

In [23]: np.insert(a, add_idx - np.arange(len(add_idx)), add_val)
Out[23]: array([1, 2, 3, 4, 5])

关于python - 将元素插入 numpy 数组的更好方法,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/52346241/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com