np.argmax() 返回最大值索引号

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)