python中numpy.squeeze函数

numpy.squeeze函数

numpy.squeeze函数的主要作用是：从数组形状中删除一维项。

用法

import numpy as np

np.squeeze(a, axis=None)

参数

• a：数组，输入数据；
• axis：None或int或tuple of ints，可选参数。

实例

实例1

1.生成数组

x = np.array([[[0,1,2]]])
x.shape

输出：

(1, 1, 3)

2.使用squeeze函数，并查看输出

a = np.squeeze(x)
a

输出：

array([0, 1, 2])

3.查看shape

a.shape

输出：

(3,)

上述例子中的a起初有3个维度，其中0轴和1轴都只有一个元素，使用squeeze函数函数后，删除了0轴和1轴两个维度。

实例2

1.生成数组

x = np.array([[[0],[1],[2]],[[3],[4],[5]]])
x.shape

输出：

(2, 3, 1)

2.使用squeeze函数，并查看输出

a = np.squeeze(x)
a

输出：

array([[0, 1, 2],
       [3, 4, 5]])

3.查看shape

a.shape

输出：

(2, 3)

上述例子中的a起初有3个维度，其中2轴只有一个元素，使用squeeze函数函数后，删除了2轴这个维度，但原来2轴中的元素并没有删除，而是合并之后成为了1轴中的元素。