您的位置:首页 > 编程语言 > Python开发

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

2017-10-08 19:56 399 查看
下面的代码需要注意return返回值的缩进,如果和for循环是同一级就会返回这个一维数组的最大下标。
import numpy as np

def 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)
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息