这里来记录一下numpy.sum()函数参数axis的问题。
代码如下：
import numpy as np
array = np.array([[1, 1, 1, 1],
[2, 2, 2, 2]])
sum_x = np.sum(array, axis=0)
sum_y = np.sum(array, axis=1)
print("The value of sum_x is: ")
print(sum_x)
print("The value of sum_y is: ")
print(sum_y)
"""
The value of sum_x is:
[3 3 3 3]
The value of sum_y is:
[4 8]
"""
可以看到，axis=0表示沿着y轴方向加和所有元素，或者说保留x轴向的元素。
axis=1表示沿着x轴方向加和所有元素，或者说保留y轴向的元素。类似的可以查看这篇numpy数组的坐标轴问题。
码字不易，如果大家觉得有用，请高抬贵手给一个赞让我上推荐让更多的人看到吧~
