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)