Python----基于numpy快速查找多个数组相同(或不同)元素的索引位置

Python----基于numpy快速查找多个数组相同(或不同)元素的索引位置

方法如下,设a与b为相同size的数组

import numpy as np
np.random.seed(2) # 设置随机种子,保证同一台电脑多次运行生成的随机数是一样的(顺便吐槽一下伪随机数生成器)
a = np.random.uniform(-1,1,(3,4))
b = np.random.uniform(-1,1,(3,4))
temp = a - b
idx = np.where(temp != 0)  # 用于查找不同元素的索引位置
idx_ = np.where(temp == 0)  # 用于查找相同元素的索引位置

"""
人生苦短,我用Python!
"""

你可能感兴趣的:(代码,python,numpy)