下面的代码需要注意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)