python报数组越界,python数组过滤越界

我尝试在

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])

如果我不在正确的轨道上,我怎么能这样做呢？

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

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

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