NumPy numpy.stack 函数

• 

  简述

  此函数沿新轴连接数组序列。从 NumPy 版本 1.10.0 开始添加此功能。需要提供以下参数。

  注意− 此函数在1.10.0及以上版本中可用。

  
  numpy.stack(arrays, axis)
  

  复制

  参数说明

  序号	参数及说明
  1	arrays 相同形状的数组序列
  2	axis 结果数组中的轴，输入数组沿该轴堆叠

• 

  例子

  
  import numpy as np 
  a = np.array([[1,2],[3,4]]) 
  print 'First Array:' 
  print a 
  print '\n'
  b = np.array([[5,6],[7,8]]) 
  print 'Second Array:' 
  print b 
  print '\n'  
  print 'Stack the two arrays along axis 0:' 
  print np.stack((a,b),0) 
  print '\n'  
  print 'Stack the two arrays along axis 1:' 
  print np.stack((a,b),1)
  

  复制

  它应该产生以下输出 -

  
  First array:
  [[1 2]
   [3 4]]
  Second array:
  [[5 6]
   [7 8]]
  Stack the two arrays along axis 0:
  [[[1 2]
   [3 4]]
   [[5 6]
   [7 8]]]
  Stack the two arrays along axis 1:
  [[[1 2]
   [5 6]]
   [[3 4]
   [7 8]]]
  

  复制