如何固定numpy.where（）调用16000x16000矩阵

Question

我有一个16000x16000矩阵，我需要从中提取布尔掩码。 布尔掩码指示某个单元格的值是否高于THRESHOLD。

以下是相关代码段

def adj_mask( dmat ):
    '''
    Returns locations where value exceeds the threshold
    '''
    global DIST    

    return numpy.where(dmat > DIST)

我有各种矩阵，我需要进行这样的选择。但是，当我用一个大的（16000x16000）矩阵调用它时，我的整个系统似乎几乎冻结（挂起）。

请建议我如何加快计算？