计算一个二维NumPy数组中所有列的总和

  • Post category:Python

计算一个二维NumPy数组中所有列的总和,可以使用 numpy.sum 函数。

首先,需要保证导入了NumPy库,可以使用以下代码导入库:

import numpy as np

接下来可以创建一个二维NumPy数组来进行测试,示例如下:

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print(arr)
# Output: 
# [[1 2 3]
# [4 5 6]
# [7 8 9]]

当以上代码运行后,可以得到一个3*3的二维数组。接下来使用 numpy.sum 函数来计算每一列的总和。在使用numpy.sum函数时,需要设置 axis 参数来指定对哪个维度求和。如果要对每一列求和,那么需要设置 axis=0。

示例如下:

col_sum = np.sum(arr, axis=0)
print(col_sum)
# Output:
# [12 15 18]

以上代码中,使用 np.sum 函数计算了 arr 中每一列的和并保存到 col_sum 中,其中 axis=0 将对列求和。

下面再举一个实际应用的例子:

假设现在有一个学生成绩表格,表格的行代表一个学生,列表示学生所得的成绩。现在要计算每一科目的平均分。可以先将表格转换为 NumPy 数组,然后计算每列的平均值即可。

表格转换为二维数组的代码如下:

score_table = np.array(
    [[85, 90, 92, 88],
     [79, 81, 75, 82],
     [94, 89, 81, 90],
     [92, 81, 80, 88],
     [82, 92, 81, 90]]
)

接下来进行每列平均分的计算:

col_mean = np.mean(score_table, axis=0)
print(col_mean)
# Output:
# [86.4 86.6 81.8 87.6]

以上代码中,使用 np.mean 函数计算了 score_table 中每一列的均值并保存到 col_mean 中,其中 axis=0 表示计算每列的平均值。

综上所述,以上就是计算一个二维NumPy数组中所有列的总和的完整攻略。