numpy中有一个求最小值的功能,里面有一个axis参数,表示维度,这里遇到了一些问题,先新建一个多维数组:
arr = np.random.randint(1,9,(3,3,2))
arr
数组如下:
[[[2, 2],
[8, 8],
[8, 7]],
[[1, 5],
[1, 3],
[3, 2]],
[[8, 4],
[4, 6],
[6, 4]]]
再去找0维度中的最小值:
arr.min(axis = 0)
把这个数组的下标写出来:
(0,0,0)(0,0,1)
(0,1,0)(0,1,1)
(0,2,0)(0,2,1)
(1,0,0)(1,0,1)
(1,1,0)(1,1,1)
(1,2,0)(1,2,1)
(2,0,0)(2,0,1)
(2,1,0)(2,1,1)
(2,2,0)(2,2,1)
现在axis=0,表示从第0个维度入手:
看第一个元素下表(0,0,0),在数组中找出所有在第1,2个维度与第一个元素下表相同的元素
(1,0,0),(2,0,0),比较这三个元素,最小值为1
最终结果为
[[1, 2],
[1, 3],
[3, 2]]