python學習筆記(一)argmax

2021-08-16 11:15:35 字數 612 閱讀 6165

(1)針對一維陣列:返回陣列中最大值元素的索引位置,如下**所示,輸出結果為元素6的索引值:4

import numpy as np

a = np.array([3, 1, 2, 4, 6, 1])

print(np.argmax(a)) #輸出結果為4

(2)針對二維陣列:引數axis可取(0,1),預設是0,表示陣列第幾維的最大值。下面分別針對axis=0,axis=1進行舉例解釋

import numpy as np

a = np.array([[1, 5, 5, 2],

[9, 6, 2, 8],

[3, 7, 9, 1]])

print(np.argmax(a, axis=0)) #輸出結果為(1,2,2,1)

import numpy as np

a = np.array([[1, 5, 5, 2],

[9, 6, 2, 8],

[3, 7, 9, 1]])

print(np.argmax(a, axis=1)) #輸出結果為(1,0,2)

(3)針對三維的陣列,推薦一篇博文  

Python學習 學習筆記(一)

python是什麼?人們為和使用python python的缺點 如今誰在使用python 流行的p2p檔案分享系統bitjorrent是乙個python程式。eve online這款大型多人網路遊戲 massively multiplayer online game,mmog 廣泛地使用pytho...

python學習學習筆記一

1,python 是完全物件導向的語言。在python中一切都是物件,函式 模組 字串等都是物件。2,資料型別 數字,字串,列表,元組,字典 數字型 整型 浮點型 布林型 非零即真 複數型 int x float x 型別轉換 非數字型 字串 列表 元祖 字典 list 元祖 元祖轉列表 tuple...

Python學習筆記 一

python學習筆記 一 關鍵知識點 1 程式列印輸出使用print語句 2 使用print輸出字串時,字串內容不帶引號。而使用字串變數名輸出時,字串內容由引號括起來 3 在python 解析器中下劃線 表示最後乙個表示式的值 4 重定向輸出符合為 5 程式中需要輸入時,實用raw input 內建...