Python學習-numpy高級函數之sort-argmax-argmin函數

Numpy 高級函數一(sort、argmax、argmin)函數使用

首先導入numpy庫

#導入numpy
import numpy as np
  1.  numpy.sort()排序函數使用,次函數對數組中的元素進行排序
#sort 排序函數
a=np.array([[6,5],[8,9]])
print("a")
print(a)

輸出結果:

a
[[6 5]
 [8 9]]

對數組a進行排序:

print("調用sort函數排序")
b=np.sort(a)
print(b)

輸出結果:

調用sort函數排序
[[5 6]
 [8 9]]

2.numpy.argmax()函數使用,返回數組中最大值的索引,三種使用方式。

定義源數組:

#argmax函數,返回數組最大值的索引
a=np.array([[2,2,1],[4,6,5],[9,8,7]])
print("a")
print(a)
  • 直接調用,返回所有元素最大值索引
print("調用argmax函數,直接調用返回最大值索引")
b=np.argmax(a)
print(b)

輸出結果:

調用argmax函數,返回最大值索引
6
  • 沿着0軸方向調用,輸出每一行最大值索引
print("沿着0軸方向調用argmax")
print(np.argmax(a,axis=0))

輸出結果:

沿着0軸方向調用argmax
[2 2 2]
  • 沿着1軸方向調用,輸出每一列最大值索引
print("沿着1軸方向調用argmax")
print(np.argmax(a,axis=1))

輸出結果:

沿着1軸方向調用argmax
[0 1 0]

3.umpy.argmin()函數使用,返回數組中最小值索引,同argmax一樣,也有直接調用,沿0軸調用和沿1軸調用三種調用方式;我這裏就不一一演示了,直接把調用代碼貼出。

首先定義一個用於操作的源數組:

#argmin函數,返回數組最小值索引
a=np.array([[1,2,4],[8,7,6],[0,3,9]])
print("a")
print(a)

下面是輸出結果:

a
[[1 2 4]
 [8 7 6]
 [0 3 9]]

調用函數

print("直接調用argmix函數")
b=np.argmin(a)
print(b)

輸出結果:
直接調用argmix函數
6


//沿0軸調用
print("沿着0軸方向調用argmin")
print(np.argmin(a,axis=0))

輸出結果:
沿着0軸方向調用argmin
[2 0 0]


//沿1軸調用
print("沿着1軸方向調用argmin")
print(np.argmin(a,axis=1))


輸出結果:
沿着1軸方向調用argmin
[0 2 0]

 

發佈了15 篇原創文章 · 獲贊 21 · 訪問量 6085
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章