import numpy as np
a = np.array([96, 85, 74, 63, 521, 41])
print(np.argmax(a)) # 取出a中元素最大值所对应的索引
c = np.array([[96, 85, 74, 63],
[36, 25, 14, 58],
[78, 98, 56, 45]])
print(np.argmax(c, axis=1))
print(np.argmax(c, axis=0)) # a[0]按列方向搜索最大值
np.argmax()
于 2021-06-17 17:09:38 首次发布