要查找NumPy数组中所有元素的和或积,可以使用NumPy中的sum和prod函数。
查找数组元素的和:
import numpy as np
# 创建一个包含5个元素的一维数组
arr = np.array([1, 2, 3, 4, 5])
# 计算数组元素的和
sum = np.sum(arr)
print(sum) # 输出 15
解释:np.sum函数用于计算数组中所有元素的和,传入数组参数后返回计算结果。
查找数组元素的积:
import numpy as np
# 创建一个包含2行3列的二维数组
arr = np.array([[1, 2, 3], [4, 5, 6]])
# 计算数组元素的积
prod = np.prod(arr)
print(prod) # 输出 720
解释:np.prod函数用于计算数组中所有元素的积,传入数组参数后返回计算结果。
需要注意的是,当数组中存在0时,计算积的结果总是0。因此,如果数组中存在0,需要在计算之前判断数组中是否存在0。
示例说明1:
import numpy as np
# 创建一个包含4个元素的一维数组
arr = np.array([2, 4, 0, 6])
# 判断数组中是否存在0
if 0 in arr:
print("数组中存在0,无法计算积")
else:
prod = np.prod(arr)
print(prod)
输出结果为”数组中存在0,无法计算积”。因为数组中存在0,无法计算积。
示例说明2:
import numpy as np
# 创建一个包含2行3列的二维数组
arr = np.array([[1, 2, 3], [4, 5, 6]])
# 计算数组元素的积
prod = np.prod(arr)
print(prod) # 输出 720
输出结果为720,因为数组中没有0,可以计算积。