在 NumPy 中移除长度为一的轴(指定轴)


要移除长度为一的轴,可以使用 NumPy 中的 **ma.MaskedArray.squeeze()** 方法。轴可以通过 "axis" 参数设置。该轴选择形状中长度为一的条目子集。如果选择了一个形状条目大于一的轴,则会引发错误。

该函数返回输入数组,但移除了所有或一部分长度为 1 的维度。这始终是 a 本身或 a 的视图。请注意,如果所有轴都被压缩,则结果是一个 0 维数组,而不是一个标量。

步骤

首先,导入所需的库 -

import numpy as np
import numpy.ma as ma

使用 numpy.array() 方法创建包含整数元素的数组 -

arr = np.array([[[15], [30], [45]]])
print("Array...
", arr) print("
Array type...
", arr.dtype)

获取数组的维度 -

print("
Array Dimensions...
",arr.ndim)

创建一个掩码数组,并将其中一些标记为无效 -

maskArr = ma.masked_array(arr, mask =[[0, 1, 0]])
print("
Our Masked Array
", maskArr) print("
Our Masked Array type...
", maskArr.dtype)

获取掩码数组的维度 -

print("
Our Masked Array Dimensions...
",maskArr.ndim)

获取掩码数组的形状 -

print("
Our Masked Array Shape...
",maskArr.shape)

获取掩码数组的元素数量 -

print("
Elements in the Masked Array...
",maskArr.size)

要移除长度为一的轴,可以使用 NumPy 中的 ma.MaskedArray.squeeze() 方法。轴可以通过 "axis" 参数设置 -

print("
Result...
",np.squeeze(maskArr, axis = 0)) print("
Shape...
",np.squeeze(maskArr, axis = 0).shape)

示例

# Python ma.MaskedArray - Remove axes of length one with specific axis

import numpy as np
import numpy.ma as ma

# Create an array with int elements using the numpy.array() method
arr = np.array([[[15], [30], [45]]])
print("Array...
", arr) print("
Array type...
", arr.dtype) # Get the dimensions of the Array print("
Array Dimensions...
",arr.ndim) # Create a masked array and mask some of them as invalid maskArr = ma.masked_array(arr, mask =[[0, 1, 0]]) print("
Our Masked Array
", maskArr) print("
Our Masked Array type...
", maskArr.dtype) # Get the dimensions of the Masked Array print("
Our Masked Array Dimensions...
",maskArr.ndim) # Get the shape of the Masked Array print("
Our Masked Array Shape...
",maskArr.shape) # Get the number of elements of the Masked Array print("
Elements in the Masked Array...
",maskArr.size) # To remove axes of length one, use the ma.MaskedArray.squeeze() method in Numpy # The axis is set using the "axis" parameter print("
Result...
",np.squeeze(maskArr, axis = 0)) print("
Shape...
",np.squeeze(maskArr, axis = 0).shape)

输出

Array...
[[[15]
[30]
[45]]]

Array type...
int64

Array Dimensions...
3

Our Masked Array
[[[15]
[--]
[45]]]

Our Masked Array type...
int64

Our Masked Array Dimensions...
3

Our Masked Array Shape...
(1, 3, 1)

Elements in the Masked Array...
3

Result...
[[15]
[--]
[45]]
Shape...
(3, 1)

更新于: 2022年2月3日

145 次查看

开启你的 职业生涯

通过完成课程获得认证

开始学习
广告