numpy中min函數

numpy提供的數組功能比較經常使用,NumPy中維數被稱爲軸,軸數稱爲秩。數組

import numpy as npspa

好比a = np.array([[1, 5, 3], [4, 2, 6]])import

a.min()返回的就是a中全部元素的最小值numpy

a.min(0)返回的就是a的每列最小值im

a.min(1)返回的是a的每行最小值

光這麼說可能有點犯迷糊,下面舉一個三維的例子

b = np.array([[[1, 2, 3], [4, 5, 6]], [[2, 3, 4], [3, 65, 1]], [[1, 33, 2], [44, 55, 66]]])

打印一下b以下

 

array(

[[[ 1, 2, 3],
[ 4, 5, 6]],

 

[[ 2, 3, 4],
[ 3, 65, 1]],

 

[[ 1, 33, 2],
[44, 55, 66]]])

b.min(k)就是b.min(axis=k),就是在他的第k個軸上投影求最小

好比b.min(0)就是

[[1 2 2],

[3, 5, 1]] ,

第一個元素1是經過比較b[0][0][0]、b[1][0][0]和b[2][0][0]獲得的

第二個元素2是經過比較b[0][0][1]、b[1][0][1]和b[2][0][1]獲得的

。。。

以上能夠當作以第一位元素爲軸投影,依次求他們的最小

若求b.min(2)就是獲得

array(

[[ 1, 4],
[ 2, 1],
[ 1, 44]])

第一個元素時經過b[0][0][0]、b[0][0][1]和b[0][0][2]獲得

。。。

比較的時候前面元素位置都不變,最後一位依次變化,找最小

一樣,max也是相似

相關文章
相關標籤/搜索