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]])