计算2D NumPy数组的所有列的总和

在NumPy中，可以使用sum函数计算2D数组的所有列的总和。首先，让我们导入NumPy模块并创建一个2D数组：

import numpy as np

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

我们将使用sum函数计算数组的每一列的总和。sum函数的axis参数指定在哪个方向上执行求和操作。使用axis = 0参数，我们可以计算每一列的总和。

# 计算2D数组的每一列的总和
col_sum = arr.sum(axis=0)

# 打印每列的总和
print(col_sum)

输出结果：

[12 15 18]

这意味着，第一列的总和为12，第二列为15，第三列为18。

完整代码

import numpy as np

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

# 计算2D数组的每一列的总和
col_sum = arr.sum(axis=0)

# 打印每列的总和
print(col_sum)

结果：

[12 15 18]

总结：使用sum函数和axis = 0参数，可以轻松地计算2D NumPy数组的每一列的总和。