以下是关于“numpy之sum()的使用及说明”的完整攻略。
在NumPy中,sum()函数是用于计算中元素的总和的函数。在本攻略中,我们介绍如何使用sum()函数来计算数组中元素的总和。
以下是示例,展示何使用sum()函数计算一维数组中元素的总和:
import numpy as np
a = np.array([1, 2, 3, 4, 5])
sum = np.sum(a)
print(sum)
输出结果为:
15
在上述代码中,我们使用np.sum()计算数组a中元素的总和,使用print()函数打印结果。
以下是一个示例,展示如何使用sum()函数计算二维中元素的总和:
import numpy as np
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
sum = np.sum(a)
print(sum)
输出结果为:
45
在上述代码中,我们使用np()函数计算二维数组a中元素的总和,并使用print()函数打印结果。
sum()函数有以下参数:
以下是两个示例,展示如何使用sum()函数计算二维数组和三维数组中元素的总和:
import numpy as np
# 示例1:计算二维数组中每列的总和
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
col_sum = np.sum(a, axis=0)
print(col_sum)
# 示例2:计算三维数组中每个二维数组的总和
b = np.array([[[1, 2, 3], [4, 5, 6], [7, 8, 9]], [[10, 11, 12], [13, 14, 15], [16, 17, 18]]])
sum = np.sum(b, axis=(1, 2))
print(sum)
输出结果为:
# 示例1输出结果
[12 15 18]
# 示例2输出结果
[ 45 126]
示例1中,我们使用sum()函数计算二维数组a中每列的总和,并使用axis=0参数指定计算每列的总和。在示例2中,我们使用sum()函数计算三维数组b中每个二维数组的总和,并使用axis=(1, 2)参数指定计算每个二维数组总和。
在使用sum()函数计算数组中元素的总和时,需要注意以下几点:
综上所述,“numpy之sum()的使用及说明”的攻略介绍了如何使用sum()函数计算数组中元素的总和。可以根据需要选择适合的函数。
本文链接:http://task.lmcjl.com/news/16887.html