np.argmax() 返回最大值索引号
ref
# -*- coding: utf-8 -*-
import numpy as np
a = np.array([[2, 4, 6, 1], [1, 5, 2, 9], [3, 7, 4, 6]])
print(a)
print(a.argmax()) # 返回整个维度的最大值索引
# 竖着比较,返回行号,有4列,故返回1*4的list
print(a.argmax(axis=0))
# 横着比较,返回列号,有3行,故返回1*3的list
print(a.argmax(axis=1))
print('-=' * 30)
tempList = a.argmax(axis=1)
for i, index in enumerate(tempList):
a[i,index] = 0
print(a[i,index])
print(a)
# 横着比较,返回列号,有3行,故返回1*3的list
print(a.argmax(axis=1))
a = np.array([[2, 4, 6, 1], [1, 5, 2, 9], [3, 7, 4, 6]])
print('=-' * 30)
print(a)
print("a.argmax(axis=1) =", a.argmax(axis=1))
print([row for row in range(a.shape[0])])
hughzip = list(zip([row for row in range(a.shape[0])], a.argmax(axis=1)))
for zipShape in hughzip:
# print(i)
a[zipShape] = np.min(a[zipShape[0],:])
print(hughzip)
print('a = \n')
print(a)