python里无论是基础的数学工具包numpy、pandas还是其他机器学习库、深度学习框架,都不可避免地会遇到选取维度运算的问题:
通过axis来指定。
- axis = 0
按行计算,得到列的性质。 - axis = 1
按列计算,得到行的性质。
以下用numpy的例子来说明一下:
import numpy as np
matrix = np.arange(0, 12).reshape(3, 4)
print(matrix)
# [[ 0 1 2 3]
# [ 4 5 6 7]
# [ 8 9 10 11]]
print(np.sum(matrix, axis=0))
# [12 15 18 21]
print(np.sum(matrix, axis=1))
# [ 6 22 38]