关键词

详解Numpy sum()(返回数组元素的总和)函数的作用与使用方法

Numpy的sum()函数是用于计算数组中所有元素的总和的函数。它可以接受一个参数,用于指定沿哪个轴计算和。它可以返回一个轴向和,也可以返回一个标量和。

下面是sum()函数的语法:

numpy.sum(arr, axis=None, dtype=None, out=None, keepdims=False, initial=0, where=True)

参数说明

  1. arr:数组
  2. axis:沿哪个轴计算和,默认为None,计算所有元素的和
  3. dtype:返回数组的数据类型,默认为None,返回输入数据的数据类型
  4. out:输出结果的数组,默认为None,返回一个新的数组
  5. keepdims:保持维度与输入数据相同
  6. initial:求和的起始值,默认为0
  7. where:开启遮盖计算,默认值为True

下面是一些使用sum()函数的实例:

例1:计算一维数组的元素总和

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
print("数组元素:", arr)
print("元素总和:", np.sum(arr))

输出结果为:

数组元素: [1 2 3 4 5]
元素总和: 15

例2:计算二维数组(矩阵)每行的元素总和

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print("数组元素:")
print(arr)

# 沿着每行计算元素的总和
row_sum = np.sum(arr, axis=1)
print("每行的元素总和:")
print(row_sum)

输出结果为:

数组元素:
[[1 2 3]
 [4 5 6]
 [7 8 9]]
每行的元素总和:
[ 6 15 24]

总结

以上就是关于Numpy sum()函数的作用和使用方法的详细攻略,sum()函数是一个用于计算数组中所有元素总和的函数,它可以接受一个参数用于指定计算轴向和。通过使用sum()函数,我们可以轻松地计算数组中所有元素的总和。

本文链接:http://task.lmcjl.com/news/17076.html

展开阅读全文