numpy.argmax

• 用例:
  numpy.argmax(a, axis=None, out=None)

• 功能:
  返回數組沿著某一條軸最大值的索引。

• 參數

變量名數據類型功能

a	數組型變量	輸入數據。
axis	整型，可選參數	默認情況下，會先將數組變為一維數組，然后求最大值的索引。否則沿著指定軸計算最大值索引。
out	數組，可選參數	若提供此參數，結果將保存到這個數組中。必須為其指定合適的形狀和數值類型。

• 返回值

變量名數據類型功能

index_array	n維數組或整數	數組中的索引。其維度與a.shape中沿軸方向的大小一致。

• 備注
  若出現多個最大值，則返回最大值第一次出現位置的索引。

• 示例:

import numpy as np a = np.arange(6).reshape(2,3) print('a數組：\n{}'.format(a)) print('a數組中最大值的索引：{}'.format(np.argmax(a))) print('a數組沿列方向最大值的索引：{}'.format(np.argmax(a, axis=0))) print('a數組沿行方向最大值的索引：{}'.format(np.argmax(a, axis=1))) b = np.arange(6) b[1] = 5 print('b數組：\n{}'.format(b)) print('b數組最大值的索引：{}'.format(np.argmax(b))) # 僅僅返回5第一次出現的位置

a數組：
[[0 1 2]
[3 4 5]]
a數組中最大值的索引：5
a數組沿列方向最大值的索引：[1 1 1]
a數組沿行方向最大值的索引：[2 2]
b數組：
[0 5 2 3 4 5]
b數組最大值的索引：1

---

github鏈接
https://github.com/wzy6642/numpy-translate

總結

以上是生活随笔為你收集整理的numpy.argmax详解的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。