查找NumPy数组元素的和与积

要查找NumPy数组元素的和与积，可以使用NumPy库中的sum()和prod()函数。

sum()函数

sum()函数用于计算数组的元素之和。可以用以下方式调用：

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
arr_sum = np.sum(arr)

print("数组的元素之和为：", arr_sum)

输出结果为：

数组的元素之和为： 15

sum()函数还可以接受axis参数，用于指定计算的轴。例如：

import numpy as np

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

# 对列进行求和
arr_sum = np.sum(arr, axis=0)

print("列的元素之和为：", arr_sum)

输出结果为：

列的元素之和为： [4 6]

prod()函数用于计算数组的元素之积。可以用以下方式调用：

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
arr_prod = np.prod(arr)

print("数组的元素之积为：", arr_prod)

输出结果为：

数组的元素之积为： 120

prod()函数也可以接受axis参数，用于指定计算的轴。例如：

import numpy as np

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

# 对列进行求积
arr_prod = np.prod(arr, axis=0)

print("列的元素之积为：", arr_prod)

输出结果为：

列的元素之积为： [3 8]

通过以上示例可以看到，sum()和prod()函数是非常方便的用于计算数组元素之和与积的工具，同时也很容易理解和使用。

Tags: Python