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

2022-10-09 12:05:21
import numpy as np
a = np.array([[2, 4, 6, 1], [1, 5, 2, 9]])
print(np.argmax(a))
print(np.argmax(a, axis=0))  #竖着比较,返回行号
print(np.argmax(a, axis=1))  #横着比较,返回列号

输出:

7
[0 1 0 1]
[2 3]

  • 作者:rainie1003
  • 原文链接:https://blog.csdn.net/xykimred/article/details/84872897
    更新时间:2022-10-09 12:05:21