Python快速找到两个数组相等元素和不相等元素的下标位置

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/sinat_29957455/article/details/84724889

找到两个两个数组下标相等和不相等的下标位置，主要借助Python的第三方库numpy来实现，numpy作为Python的第三方库还是非常高效的，而且内置了许多的函数，方便使用。

import numpy as np

if __name__ == "__main__":
    a = np.array([1,2,3,4,5])
    b = np.array([1,2,3,3,4])
    #定义数组下标
    index = np.arange(0,5)
    #找到两个数组相等元素的下标位置
    print(index[a == b])
    #找到两个数组不相等元素的下标位置
    print(index[a != b])

[0 1 2]
[3 4]