按条件索引大型2d numpy数组

Question

我有一个2D numpython数组，即（15003712）。我想找出数组中值在-10到-40之间的指示符。到目前为止，我已经：

for item in lon_array:
    for value in item:
        if value >= -40. and value <= -10:
            find_index =  np.where(lon_array == value)
            index =  np.asarray(find_index).T

因为它是一个非常大的数组，有什么方法可以让它更快？你知道吗

Answer 1

如果lon_array是一个列表列表（Python的内置基本数据类型），那么使用enumerate(...)将是了解元素索引的最佳方法：

for y, row in enumerate(lon_array):
    for x, value in enumerate(row):
        if -40 <= value <= -10:
            index = (y, x)
            # do something useful with it...