NumPy numpy.ndarray.flat 函数

• 

  简述

  此函数返回数组上的一维迭代器。它的行为类似于 Python 的内置迭代器。
• 

  例子

  
  import numpy as np 
  a = np.arange(8).reshape(2,4) 
  print 'The original array:' 
  print a 
  print '\n' 
  print 'After applying the flat function:' 
  # returns element corresponding to index in flattened array 
  print a.flat[5]
  

  它的输出如下 -

  
  The original array:
  [[0 1 2 3]
   [4 5 6 7]]
  After applying the flat function:
  5