在numpy中按数字求和数组

在numpy中，可以使用numpy.sum()函数按数字求和数组。

numpy.sum()函数用于计算数组中元素的总和。它可以接受一个数组作为参数，并返回数组中所有元素的总和。

以下是numpy.sum()函数的使用示例：

import numpy as np

# 创建一个数组
arr = np.array([1, 2, 3, 4, 5])

# 求和数组中的所有元素
result = np.sum(arr)

print(result)  # 输出：15

在上述示例中，我们首先导入了numpy库，并创建了一个包含整数的数组arr。然后，我们使用np.sum()函数对数组中的所有元素进行求和，并将结果存储在变量result中。最后，我们打印出结果，得到了数组中所有元素的总和。

numpy.sum()函数还可以接受额外的参数，用于指定求和的轴。通过指定轴，我们可以沿着特定的维度对数组进行求和。例如，对于多维数组，可以通过指定axis参数来沿着行或列进行求和。

以下是一个使用axis参数的示例：

import numpy as np

# 创建一个二维数组
arr = np.array([[1, 2, 3], [4, 5, 6]])

# 沿着行求和
row_sum = np.sum(arr, axis=0)

# 沿着列求和
column_sum = np.sum(arr, axis=1)

print(row_sum)  # 输出：[5 7 9]
print(column_sum)  # 输出：[6 15]

在上述示例中，我们创建了一个二维数组arr。然后，我们使用np.sum()函数分别沿着行和列对数组进行求和，并将结果存储在row_sum和column_sum变量中。最后，我们打印出结果，得到了沿着行和列的求和结果。

总结起来，numpy.sum()函数是一个用于计算数组元素总和的强大工具，可以灵活地应用于不同维度的数组。在数据分析、科学计算等领域中，它经常被用于对数据进行统计和分析。在使用numpy.sum()函数时，我们可以根据需要指定轴参数，以满足不同的求和需求。

腾讯云相关产品和产品介绍链接地址：