numpy.sum()坐标轴问题

这里来记录一下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数组的坐标轴问题。

码字不易，如果大家觉得有用，请高抬贵手给一个赞让我上推荐让更多的人看到吧~