博客

Python中axis的意思：

就记住axis=i，操作就是沿第i维变化的方向进行;

对于一个(5,4,3,2)的数组data：dim 0 : 5 ; dim 1 : 4; dim 2 : 3; dim 3 : 2

axis=0,操作时只有第0维的下标变化其他不变，操作结束后变为(4,3,2)

axis=1,操作时只有第1维的下标变化其他不变，操作结束后变为(5,3,2)

axis=2,操作时只有第2维的下标变化其他不变，操作结束后变为(5,4,2)

axis=3,操作时只有第3维的下标变化其他不变，操作结束后变为(5,4,3)

data = np.random.randint(0,5,[5,4,3,2])

a = np.max(data, axis = 0, keepdims = False)

a = np.max(data, axis = 0, keepdims = True)