python - 如何在某些索引处复制 numpy 数组值

Question

我有一个 numpy 数组，我想将某些索引处的部分数组复制到另一个数组。

arr = np.arange(10)
np.random.shuffle(arr)
print arr
[0 3 4 2 5 6 8 7 9 1]

我想复制索引处的值

copy_indices = [3, 7, 8]

有什么好的方法可以做到这一点吗？

Answer 1

使用这种方法怎么样？

In [16]: arr
Out[16]: array([2, 9, 5, 6, 1, 4, 7, 8, 3, 0])

In [17]: copy_indices
Out[17]: [3, 7, 8]

In [18]: sliced_arr = np.copy(arr[copy_indices, ])

# alternatively
# In [18]: sliced_arr = arr[copy_indices, ]

In [19]: sliced_arr
Out[19]: array([6, 8, 3])

---

PS：高级索引（如此处）实际上返回数组的副本。所以，使用np.copy()是可选的。