Python中 axis=0、axis=1是行还是列?

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]

你可能感兴趣的:(#,Python编程问题,python,numpy,机器学习)