python - 根据索引列表从numpy数组中获取和替换值

Question

是否可以根据索引列表(例如 1 和 3)从 numpy 数组中获取值？然后，我想放置另一个值而不是它们。

这是示例:

import numpy as np

array = np.array([0, 1, 2, 8, 4, 9, 1, 2])
idx = [3, 5]

所以我想用另一个值替换 X[3] = 8 和 X[5] = 9 ，但我不想在循环中这样做，因为我可以有大数组。它是一种方法还是一个函数来执行这样的操作但不是在循环中？

Answer 1

使用 np.r_ :

larr = np.array([0, 1, 2, 8, 4, 9, 1, 2])
larr[np.r_[3, 5]]

输出

array([8, 9])

正如@MadPhysicist 所建议的，使用 larr[np.array([3, 5])]也会工作，而且速度更快。