Python NumPy compress方法用法及代码示例

例子

压缩一维数组

np.compress([True, False, True, False], [4,5,6,7])



array([4, 6])

请注意，这相当于以下内容：

a = np.array([4,5,6,7])
mask = [True, False, True, False]
a[mask]



array([4, 6])

截断

当条件数组的大小小于输入数组的大小时，就会发生截断：

a = np.array([4,5,6,7])
np.compress([True, False], a)



array([4])

这里，仅考虑数组中的前两个值 - 但仅返回第一个值，因为第二个值被标记为 False。

压缩二维数组

考虑以下二维数组：

a = np.array([[4,5],[6,7]])
a



array([[4, 5],
       [6, 7]])

从展平数组中提取

np.compress([True, False, False, True], a)



array([4, 7])

逐行提取

np.compress([True, False], a, axis=0)



array([[4, 5]])

按列提取

np.compress([True, False], a, axis=1)



array([[4],
       [6]])