numpy ravel、squeeze函数

1、ravel函数

函数原型

numpy.ravel(a, order='C')

函数说明

ravel函数将多维数组变成一维数组。

函数使用

>>> a = np.reshape(np.arange(12), (3, 4))
>>> a
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])
>>> b = np.ravel(a)
>>> b
array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11])

2、squeeze函数

函数原型

numpy.squeeze(a, axis=None)

函数说明

squeeze函数压缩数组的冗余的维度，比如有一个shape为（2，1，4）的数组，第二个维度为1，这一个维度就可以认为是冗余的，此时就可以进行压缩，成为shape为（2，4）的数组。

函数使用

>>> a = np.zeros((2, 1, 4))
>>> a.shape
(2, 1, 4)
>>> b = np.squeeze(a, axis=1)
>>> b.shape
(2, 4)