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

少云清

于 2022-08-31 21:24:57 发布

阅读量3.3k

点赞数 1

分类专栏： # Python编程问题文章标签： python numpy 机器学习

本文链接：https://blog.csdn.net/brawly/article/details/126632554

版权

Python编程问题专栏收录该内容

2 篇文章

订阅专栏

本文主要介绍了在Python中使用numpy和pandas库时遇到的维度选取问题。通过实例展示了axis=0和axis=1分别对应列计算和行计算的操作，如numpy.sum()函数的用法，帮助读者理解数组操作中的轴向概念。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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]