numpy中sum(axis=1)与sum(axis=0)的区别

最新推荐文章于 2024-06-13 14:53:48 发布

原创最新推荐文章于 2024-06-13 14:53:48 发布 · 7.9k 阅读

20 ·

CC 4.0 BY-SA版权

文章标签：

#numpy

机器学习专栏收录该内容

4 篇文章

订阅专栏

本文通过实例演示了如何使用Python的NumPy库进行数组的创建及基本操作，重点介绍了按行（axis=1）和按列（axis=0）进行求和的方法。

Python3.9

Conda

Python

Python 是一种高级、解释型、通用的编程语言，以其简洁易读的语法而闻名，适用于广泛的应用，包括Web开发、数据分析、人工智能和自动化脚本

import numpy
a = numpy.array([
    [[5,10,15],
    [10,20,30],
    [20,40,60]],
    
    [[5,10,15],
    [10,20,30],
    [20,40,60]]
])
#按行相加，此时第一、二行都是[5,10,15],[10,20,30],[20,40,60]
b = a.sum(axis=1)
#按列相加，此时第一列是[5,10,15],第二列是[10,20,30]，第三列是[20,40,60]
c = a.sum(axis=0)

print(b)
print(c)

结果：

[[ 35  70 105]
 [ 35  70 105]]
[[ 10  20  30]
 [ 20  40  60]
 [ 40  80 120]]

总结：简单的理解，axis=1按行的方向相加，返回每个行的值；axis=0按列相加，返回每个列的值。

您可能感兴趣的与本文相关的镜像

Python3.9

Conda

Python

Python 是一种高级、解释型、通用的编程语言，以其简洁易读的语法而闻名，适用于广泛的应用，包括Web开发、数据分析、人工智能和自动化脚本