Python numpy Arrayterator.flat用法及代碼示例

本文簡要介紹 python 語言中 numpy.lib.Arrayterator.flat 的用法。

用法:

property  lib.Arrayterator.flat

Arrayterator 對象的一維平麵迭代器。

該迭代器返回要迭代的數組元素數組生成器逐個。它類似於numpy.flatiter.

例子：

>>> a = np.arange(3 * 4 * 5 * 6).reshape(3, 4, 5, 6)
>>> a_itor = np.lib.Arrayterator(a, 2)

>>> for subarr in a_itor.flat:
...     if not subarr:
...         print(subarr, type(subarr))
...
0 <class 'numpy.int64'>