python3实现一维数组求最大值返回参数需要注意的地方

来源:互联网 发布:abaqus软件 编辑:程序博客网 时间:2024/06/03 19:48
下面的代码需要注意return返回值的缩进,如果和for循环是同一级就会返回这个一维数组的最大下标。
import numpy as npdef getMax1(array):    if len(array)<1:#len(array)得到列表长度        return [-1,-1]    index = 1    unit = array[index]    for j in range(1,len(array)):        if array[j] > array[index+1] and array[j] > array[index-1]:#if的含多条件语句的写法            unit = array[j]            return [unit,j]#返回的是一个列表类型,最大值和它的坐标,注意此处是在for循环里的if下得到最大值的j就返回。def getMaxInCol(matrix, colNum):    array = [col[colNum] for col in matrix]    return getMax1(array)print("please row:")n = int(input())print("please column:")m = int(input())matrix = np.random.randint(100,999,size=[n,m])print(matrix)test = getMaxInCol(matrix,(len(matrix[0])-1)//2)print(test)



原创粉丝点击