axis = 0,对横轴(第0维)进行操作,运算的方向为纵向运算。即求列和、最大、最小等
axis = 1,对纵轴(第1维)进程操作,运算的方向为横向运算。即求行和、最大、最小等
import numpy as np
import numpy as np
data = np.array([[1,1,1,1],
[2,2,2,2],
[3,3,3,3],
[4,4,4,4]])
Out[1]:
array([[1, 1, 1, 1],
[2, 2, 2, 2],
[3, 3, 3, 3],
[4, 4, 4, 4]])
#对横轴进行操作,即纵向运算
np.sum(data,axis=0)
Out[2]:
array([10, 10, 10, 10])
#对纵轴进行操作,即横向运算
np.sum(data,axis=1)
Out[3]:
array([ 4, 8, 12, 16])
参考文献:
https://blog.youkuaiyun.com/brucewong0516/article/details/79030994
本文详细解析了NumPy中axis参数的功能,展示了如何通过axis=0和axis=1对矩阵进行纵向和横向的求和、最大值及最小值运算。通过具体示例,帮助读者理解在数据处理中如何有效利用这一特性。
9188

被折叠的 条评论
为什么被折叠?



