python数组过滤越界

Question

我尝试在 python 中执行此操作，但出现错误:

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

这导致:

IndexError: index 5 is out of bounds for axis 0 with size 5

什么给了？我认为我在这里做的操作是正确的。

我希望如果我这样做

array_to_filter[equal_array]

它会回来

np.array([1,2,5])

如果我不在正确的轨道上，我将如何做到这一点？

Answer 1

在最后一条语句中，数组的索引是 1、2、5、5 和 5。索引 5 指的是数组中的第 6 个元素，而您只有 5 个元素。 array_to_filter[5] 不存在。

[i for i in np.unique(equal_array) if i in array_to_filter]

会返回你想要的答案。如果它也存在于 array_to_filter 中，则返回 equal_array 中的每个唯一值