在 Numpy 中移除长度为 1 且指定轴的轴
numpyserver side programmingprogramming更新于 2025/9/4 5:37:17
要移除长度为 1 的轴，请使用 Numpy 中的 ma.MaskedArray.squeeze() 方法。轴通过"axis"参数设置。该轴会选择形状中长度为 1 的条目子集。如果选择的轴的形状条目数大于 1，则会引发错误。
该函数返回一个输入数组，但长度为 1 的所有维度或其子集已被移除。这始终是 a 本身或 a 的视图。请注意，如果所有轴都被挤压，则结果为零维数组，而不是标量。
步骤
首先，导入所需的库 −
import numpy as np import numpy.ma as ma
使用 numpy.array() 方法创建一个包含 int 元素的数组 −
arr = np.array([[[15], [30], [45]]])
print("数组...
", arr)
print("
数组类型...
", arr.dtype)
获取数组的维度 −
print("
数组维度...
",arr.ndim)
创建一个掩码数组并将其中一些掩码为无效 −
maskArr = ma.masked_array(arr, mask =[[0, 1, 0]])
print("
我们的掩码数组
", maskArr)
print("
我们的掩码数组类型...
", maskArr.dtype)
获取掩码数组的维度 −
print("
我们的掩码数组维度...
",maskArr.ndim)
获取掩码数组的维度 −
print("
我们的掩码数组形状...
",maskArr.shape)
获取掩码数组元素数量 −
print("
掩码数组中的元素...
",maskArr.size)
要移除长度为 1 的轴，请使用 Numpy 中的 ma.MaskedArray.squeeze() 方法。轴使用"axis"设置参数&减;
print("
结果...
",np.squeeze(maskArr, axis = 0))
print("
形状...
",np.squeeze(maskArr, axis = 0).shape)
示例
# Python ma.MaskedArray - 删除长度为 1 且具有特定轴的轴
import numpy as np
import numpy.ma as ma
# 使用 numpy.array() 方法创建一个包含 int 元素的数组
arr = np.array([[[15], [30], [45]]])
print("数组...
", arr)
print("
数组类型...
", arr.dtype)
# 获取数组的维度
print("
数组维度...
",arr.ndim)
# 创建一个掩码数组，并将其中一些掩码为无效
maskArr = ma.masked_array(arr, mask =[[0, 1, 0]])
print("
我们的掩码数组
", maskArr)
print("
我们的掩码数组类型...
", maskArr.dtype)
# 获取掩码数组的维度
print("
我们的掩码数组维度...
",maskArr.ndim)
# 获取掩码数组的形状
print("
我们的掩码数组形状...
",maskArr.shape)
# 获取掩码数组的元素数量
print("
掩码数组中的元素...
",maskArr.size)
# 要删除长度为 1 的轴，请使用 Numpy 中的 ma.MaskedArray.squeeze() 方法
# 使用"axis"参数设置轴
print("
结果...
",np.squeeze(maskArr, axis = 0))
print("
Shape...
",np.squeeze(maskArr, axis = 0).shape)
输出
数组... [[[15] [30] [45]]] 数组类型... int64 数组维度... 3 我们的掩码数组 [[[15] [--] [45]]] 我们的掩码数组类型... int64 我们的掩码数组维度... 3 我们的掩码数组形状... (1, 3, 1) 掩码数组中的元素... 3 结果... [[15] [--] [45]] Shape... (3, 1)
