numpy中.flat的意義是什么？

Question

這兩個代碼有何不同？ 或者根本沒有這樣的差異？

import numpy as np
a = np.array([1,2,4,5,5,6])
for item in a.flat:
    print(item)   

要么

import numpy as np
a = np.array([1,2,4,5,5,6])
for i in a:
    print(i)

兩者給出了相同的結果

1
2
4
5
5
6

Answer 1

屬性.flat將扁平化陣列。 https://docs.scipy.org/doc/numpy/reference/generated/numpy.ndarray.flat.html

np.array([[1], [2], [3], [4], [5]]).flat
>[1, 2, 3, 4, 5]

np.array([[[[[1, 2, 3, 4, 5]]]]]).flat
>[1, 2, 3, 4, 5]

np.array([1, 2, 3, 4, 5]).flat
>[1, 2, 3, 4, 5]

因為您使用的是已經平坦的數組，所以兩段代碼的輸出都是相同的。

嘗試這個：

import numpy as np
a = np.array([[1,2,4],[5,5,6]])
for item in a.flat:
    print(item)

>1
>2
>4
>5
>5
>6


import numpy as np
a = np.array([[1,2,4],[5,5,6]])
for i in a:
    print(i)

>[1 2 4]
>[5 5 6]