numpy.ndarray.flat


此函数返回数组上的1-D迭代器。它的行为类似于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