axis后面的值是指定一个轴做运算。比如:
import numpy as np
np.sum([[0, 1], [0, 5]], axis=0) # 沿行
array([0, 6])
np.sum([[0, 1], [0, 5]], axis=1) # 沿列
array([1, 5]) # 0+1;0+5
在二维情况下,具体指沿着行求和,还是沿着列求和。
axis后面的值是指定一个轴做运算。比如:
import numpy as np
np.sum([[0, 1], [0, 5]], axis=0) # 沿行
array([0, 6])
np.sum([[0, 1], [0, 5]], axis=1) # 沿列
array([1, 5]) # 0+1;0+5
在二维情况下,具体指沿着行求和,还是沿着列求和。