极客教程NumPy 数组操作 numpy.squeeze
这个函数从给定数组的形状中去除一维条目。这个函数需要两个参数。
numpy.squeeze(arr, axis)
其中，
| 编号 | 参数和描述 |
|---|---|
| 1 | arr 输入数组 |
| 2 | axis int 或者 int 元组。选择形状中的单一维度子集 |
示例
import numpy as np
x = np.arange(9).reshape(1,3,3)
print 'Array X:'
print x
print '\n'
y = np.squeeze(x)
print 'Array Y:'
print y
print '\n'
print 'The shapes of X and Y array:'
print x.shape, y.shape
它的输出如下：
Array X:
[[[0 1 2]
[3 4 5]
[6 7 8]]]
Array Y:
[[0 1 2]
[3 4 5]
[6 7 8]]
The shapes of X and Y array:
(1, 3, 3) (3, 3)