Python NumPy sum()函数完全指南：用法、示例与技巧

2 年 ago
清, 扬
1 minute

这是文章《在Python中使用numpy.sum()》的第1部分（共1部分）。

内容片段: Python的numpy.sum()函数用于沿给定轴获取数组元素的和。

Python的numpy库中sum()函数的语法

Python NumPy sum()方法的语法是：

sum(array, axis, dtype, out, keepdims, initial)
  • 数组元素用于计算总和。
  • 如果未提供轴参数，则返回所有元素的总和。如果轴是一个整数元组，则返回给定轴上所有元素的总和。
  • 我们可以指定dtype参数来指定返回的输出数据类型。
  • out变量用于指定放置结果的数组。这是一个可选参数。
  • keepdims是一个布尔参数。如果设置为True，则被缩减的轴在结果中保留为长度为一的维度。
  • initial参数指定求和的起始值。

Python numpy sum() 示例

让我们来看一些numpy.sum()函数的示例。

数组中所有元素的总和

如果我们只通过数组将其传递给 sum() 函数，它将被扁平化，并返回所有元素的总和。

import numpy as np

array1 = np.array(
    [[1, 2],
     [3, 4],
     [5, 6]])

total = np.sum(array1)
print(f'所有元素的总和为 {total}')

输出：所有元素的和为21。

2. 沿轴的数组元素之和

如果我们指定了轴值，将返回沿该轴的元素之和。如果数组形状为（X，Y），则沿0轴的求和结果形状将为（1，Y）。沿1轴的求和结果形状将为（1，X）。

import numpy as np

array1 = np.array(
    [[1, 2],
     [3, 4],
     [5, 6]])

total_0_axis = np.sum(array1, axis=0)
print(f'0轴上元素的总和为 {total_0_axis}')

total_1_axis = np.sum(array1, axis=1)
print(f'1轴上元素的总和为 {total_1_axis}')

输出：

0轴上元素的总和为 [ 9 12]
1轴上元素的总和为 [ 3  7 11]

3. 指定求和的输出数据类型

import numpy as np

array1 = np.array(
    [[1, 2],
     [3, 4]])

total_1_axis = np.sum(array1, axis=1, dtype=float)
print(f'1轴上元素的总和为 {total_1_axis}')

输出：1轴上元素的总和为[3. 7.]

4. 总和的初始值

import numpy as np

array1 = np.array(
    [[1, 2],
     [3, 4]])

total_1_axis = np.sum(array1, axis=1, initial=10)
print(f'1轴上元素的总和为 {total_1_axis}')

输出：1轴上元素的总和为[13 17] 参考：API文档

bannerAds