Python numpy ndarray.flat用法及代码示例

本文简要介绍 python 语言中 numpy.ndarray.flat 的用法。

用法:

ndarray.flat

数组上的一维迭代器。

这是一个 numpy.flatiter 实例，其行为类似于 Python 的内置迭代器对象，但不是其子类。

例子：

>>> x = np.arange(1, 7).reshape(2, 3)
>>> x
array([[1, 2, 3],
       [4, 5, 6]])
>>> x.flat[3]
4
>>> x.T
array([[1, 4],
       [2, 5],
       [3, 6]])
>>> x.T.flat[3]
5
>>> type(x.flat)
<class 'numpy.flatiter'>

赋值示例：

>>> x.flat = 3; x
array([[3, 3, 3],
       [3, 3, 3]])
>>> x.flat[[1,4]] = 1; x
array([[3, 1, 3],
       [3, 1, 3]])