**
axis=0,1,2
**
对于一些矩阵我们经常需要求取他们的平均值mean,求和sum等操作,但是楼主本身总是忘记axis=0,axis=1时的各类操作到底是行相加还是列相加,在此简单记录以下防止后面忘记。
假设我们有一个矩阵为MxN,通常我们称行为0轴,列为1轴。不知道憋人怎么理解的,楼主总是混淆,比如最简单的。如下所示:
import numpy as np
arr = np.array([[1,1,1,1],
[2,2,2,2]])
result = np.mean(arr,axis=1)
print(result)
>>>>>>>>>>>>>>>>>>>>>>>>>>>
out:[1. 2.]
我通常会这样理解,既然axis=1那就是对列求均值,因为我们通常认为列为1轴的,但是为啥结果时对一行进行求均值呢?
对此为了方便记忆,我这样理解的,对于axis=1,那么对应的也就是列,而上述arr为2X4的矩阵,其列的位置为4。所以对应的求均值的元素个数也应该是有4个元素,因此有4个元素就是一行了,因此对应每行求取均值。