numpy提供的數組功能比較常用,NumPy中維數被稱為軸,軸數稱為秩。
import numpy as np
比如a = np.array([[1, 5, 3], [4, 2, 6]])
a.min()返回的就是a中所有元素的最小值
a.min(0)返回的就是a的每列最小值
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也是類似