Python | numpy matrix.compress()

matrix.compress(condition, axis=None, out=None), 是 NumPy 库中 matrix 类的方法，用于根据给定的条件压缩 matrix 的行或列。

语法

matrix.compress(condition, axis=None, out=None)

参数

• condition：ndarray 类型或能够被转化为 ndarray 类型的对象，用于压缩 matrix 的行或列。
• axis：目标轴的维度位置，若为 None，则压缩 matrix 的所有元素并以展平的一维形式返回被压缩的元素，否则根据目标轴返回压缩后的 matrix。默认值为 None。
• out：结果矩阵。默认为 None。

【注】NumPy 库中的 ndarray 类型，指了一个元素相同的多维数组。

返回值

• 返回一个按照给定条件压缩后的 matrix。

示例

考虑以下例子：

import numpy as np

# 定义一个 3x3 的矩阵
x = np.array([[0, 1, 2], [3, 4, 5], [6, 7, 8]])

# 压缩所有行，并将结果展平为一维
y = x.compress([False, False, True], axis=0)
print(y)  # [6 7 8]

# 压缩所有列
z = x.compress([True, False, True], axis=1)
print(z)
"""
[[0 2]
 [3 5]
 [6 8]]
"""

在上面的例子中，我们定义了一个 3x3 的矩形 x，并使用 compress() 方法，压缩了矩阵的行和列。在第一个例子中，我们使用了一个只有最后一行为真的条件，因此 matrix.compress() 方法返回了最后一行。在第二个例子中，我们只压缩了第 $0$ 和 $2$ 列，因此我们得到了一个只包含第 $0$ 和 $2$ 列的矩形。