二维数组的高效查找（Python）

二维数组中的查找

题目：在一个二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。

import numpy as np
def number(array,data):
    if array == None:
        return False
    i = 0
    rows = np.shape(array)[0]  #获取行数
    cols = np.shape(array)[1] #获取列数
    j = cols-1
    while i<rows and j>=0:
        if array[i][j] == data:
            return True
        elif array[i][j] > data:
            j = j-1
        else:
            i = i+1
    return False
if __name__ == '__main__' :
    array=[
        [0, 1, 2, 3, 4],
        [10, 11, 12, 13, 14],
        [20, 21, 22, 23, 24],
        [30, 31, 32, 33, 34],
        [40, 41, 42, 43, 44]
    ]
    print(number(array,1))