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:' # 返回与展平数组中的索引对应的元素 print a.flat[5]
它的输出结果如下 −
The original array: [[0 1 2 3] [4 5 6 7]] After applying the flat function: 5