用数值c替换浮点数值a到b大数组

Question

我有一个浮点数组（2000 * 2000），我想对数字进行分类。 因此，10到20之间的所有数字都应该用15代替，20 - 60之间的数字应该用40代替，等等。

我用一些if语句编写了一些循环遍历所有行和列的东西......但它需要永远运行大型数组。有人知道如何加快速度吗？

for a in range(grid.shape[0]): #grid is an array 
    for b in range(grid.shape[1]):  
     for c in range(len(z)): 
      if z[c][0] <= grid[a][b] < z[c][1]: # z is a list containing [lower,upper,replace_value] 
       grid[a][b]=z[c][2] 

Answer 1

会是这样的工作吗？

>>> import numpy as np 
>>> grid = np.random.random((5,5)) * 100 
>>> z = np.array([0, 10, 20, 60, 100.]) 
>>> replace_value = np.array([np.nan, 5., 15., 40., 80.]) 

>>> grid = replace_value[z.searchsorted(grid)] 
>>> print grid 
[[ 15. 40. 80. 80. 15.] 
[ 80. 40. 15. 80. 80.] 
[ 15. 80. 5. 15. 40.] 
[ 40. 80. 5. 5. 80.] 
[ 40. 5. 80. 5. 40.]]