Python:比较numpy数组和不带循环的子numpy数组



我的问题很简单,但我不知道如何在没有循环的情况下解决它。

我有第一个numpy数组:

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

和子阵列(不一定以相同的方式排序(:

Sub array = np.array([8, 3, 5])

我想创建一个bool数组,该数组的大小与完整数组的大小相同,如果给定的FullArray值存在于SubArray中,则返回True,无论哪种方式都返回False。

例如,我希望得到:

BoolArray = np.array([False, False, False, True, False, True, False, False, True, False])

有没有一种方法可以在不使用循环的情况下做到这一点?

您可以使用np.isin:

np.isin(FullArray, SubArray)
# array([False, False, False, True, False, True, False, False, True, False])

相关内容

  • 没有找到相关文章

最新更新