python - 使用numpy时如何忽略索引数组中的越界值？

Question

我需要使用索引数组将一个数组分配给另一个数组。但是有些值是越界的...

a = np.array([0, 1, 2, 3, 4])
b = np.array([10, 11, 12, 13, 14])
indexes = np.array([0, 2, 3, 5, 6])

a 和 b 大小相同。如果我使用 a[indexes] = b，它会抛出一个 IndexError。我希望它忽略越界值 5 和 6，这样 a 就会变成 [10, 1, 11, 12, 4]。

我试着做 indexes[indexes > b.size()] = 0但这会弄乱索引 0 处的值。如何解决？

编辑

索引不一定是有序的。例如:

indexes = np.array([2, 3, 0, 5, 6])

a 应该变成 np.array([12, 1, 10, 11, 4])

Answer 1

您可以过滤掉那些无效的索引:

indexes = indexes[indexes < len(a)]

a[indexes] = b[indexes]

输出:

array([10,  1, 12, 13,  4])