Python中在用数学工具包numpy、pandas时,总是会出错,并且在运用深度学习框架Pytorch选取维度也会出现错误,因此特此总结如下:
- 对于维度选取问题:0轴垂直往下,1轴向右水平延伸。
- axis = 0,表示按 列 计算,按列填充
- axis = 1,表示按 行 计算,按行填充
import numpy as np
matrix = np.arange(0, 12).reshape(3, 4) # 生成12个整数,并重组成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]