numpy.argmax

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

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

• 參數(shù)

變量名數(shù)據(jù)類型功能

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

• 返回值

變量名數(shù)據(jù)類型功能

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

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

• 示例:

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

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

---

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

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。