Python | numpy matrix.sum()

简介

numpy matrix.sum()是numpy库中用于计算矩阵中所有元素之和的函数。它将沿着给定轴计算总和或所有元素的总和，具体取决于传递的参数。

语法

numpy.matrix.sum(axis=None, dtype=None, out=None, keepdims=False)

axis: 整数或整数元组，可选。要削减的轴或轴的序号。如果未传递任何值，则将总和计算为所有元素的总和。

dtype: 返回数据类型，可选，默认不更改数据类型。

out: carray，可选。用于放置结果的替代输出数组。它必须具有与期望输出相同的形状和类型。

keepdims: bool，可选。如果该值为True，则保持原始数组的维数。

示例

import numpy as np

# 创建矩阵
matrix = np.matrix([[1, 2], [3, 4]])

# 计算所有元素之和
sum_all = matrix.sum()

# 计算每一行元素之和
sum_row = matrix.sum(axis=1)

# 计算每一列元素之和
sum_col = matrix.sum(axis=0)

在上面的示例中，我们首先创建了一个 2x2 的矩阵。接下来，我们分别计算了整个矩阵中所有元素的总和、每一行元素的总和和每一列元素的总和。在不同的情况下，我们使用了不同的轴参数来指定要该计算元素总和的轴。如果我们不传递任何值，它默认为None，则会计算所有元素的总和。

>>> sum_all
10
>>> sum_row
matrix([[3],
        [7]])
>>> sum_col
matrix([[4, 6]])

总结

numpy库中的 matrix.sum() 函数是一个好用的函数，它可以用于计算矩阵中所有元素之和。此函数基于传递给它的参数来确定沿哪个轴计算元素总和。 您可以在许多使用numpy矩阵的应用程序中使用此函数。