我的问题很简单,但我不知道如何在没有循环的情况下解决它。
我有第一个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])