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也是类似