numpy.squeeze(a,axis=None)
从数组形状中移除一维条目
参数:
a – 数组形状np.array()
axis – 可以是一个整数,可以是一个元组或者None。默认为None。axis用于指定需要删除的维度,但是指定的维度必须为单维度,否则将会报错.
返回:压缩后的ndarray()
减少了维度的数据子集
>>> x = np.array([[[0], [1], [2]]])
>>> x.shape
(1, 3, 1)
#axis=None
>>> a = np.squeeze(x)
>>> a
array([0, 1, 2])
>>> a.shape #不指定轴,将全部一维的条目压缩
(3,)
#axis=Tuple
>>> b = np.squeeze(x,axis=(2,))
>>> b
array([[0, 1, 2]])
>>> b.shape
(1, 3)