Numpy 高級函數一(sort、argmax、argmin)函數使用
首先導入numpy庫
#導入numpy
import numpy as np
- 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]