Numpy元素的遍歷方法是什么

這篇文章主要講解了“Numpy元素的遍歷方法是什么”，文中的講解內容簡單清晰，易于學習與理解，下面請大家跟著小編的思路慢慢深入，一起來研究和學習“Numpy元素的遍歷方法是什么”吧！

一、單個數組的遍歷：numpy.nditer 的使用

迭代器對象 numpy.nditer 提供了一種靈活訪問一個或者多個數組元素的方式。一般情況下的使用方式如下：for x in np.nditer(data_in, order='')，可以遍歷nparray中的所有數據，舉例如下：

a = np.arange(12).reshape(3,4)
for x in np.nditer(a):
    print(x,end=',')
# 0,1,2,3,4,5,6,7,8,9,10,11,

格式：np.nditer(data_in, order='',op_flags =[],flags=[])

參數解釋如下：

1.1 參數：order

通過參數 order='F' or  order='C' 可以通過改變遍歷的順序；舉例如下：

for x in np.nditer(a, order='F'): # Fortran order，即是列序優先；for x in np.nditer(a.T, order='C'): # C order，即是行序優先；

for x in np.nditer(a,order='F'):
    print(x,end=',')
# 0,4,8,1,5,9,2,6,10,3,7,11,
for x in np.nditer(a,order='C'):
    print(x,end=',')
# 0,1,2,3,4,5,6,7,8,9,10,11,

1.2 參數：op_flags

默認情況下，nditer 對nparray進行遍歷時，遍歷出的元素為只讀對象（read-only），若我們需要修改其中的值，需要設置：op_flags=['readwrite']

a = np.arange(12).reshape(3,4)
for x in np.nditer(a,op_flags=['readwrite']):
    x[...] = 2*x
print(a)
# [[ 0  2  4  6]
#  [ 8 10 12 14]
#  [16 18 20 22]]

1.3 參數：flags

嘗試用的參數設置為：flags = ['external_loop'] ，其余的設置不常用，就不列舉了；當設置flags = ['external_loop'] 后，可以實現對二維數組的行或者列的輸出，在今后的寫代碼過程中，給大家帶來便利；舉例如下：

舉例：

# 返回索引值舉例
a = np.arange(12).reshape(3,4)
for x in  np.nditer(a,flags=['f_index'],order='F'):
    print(x,end=',')
# 0,4,8,1,5,9,2,6,10,3,7,11,
for x in  np.nditer(a,flags=['f_index'],order='C'):
    print(x,end=',')
# 0,1,2,3,4,5,6,7,8,9,10,11,

# 返回列方向的多個一維數組
a = np.arange(12).reshape(3,4)
for x in  np.nditer(a,flags=['external_loop'],order='F'):
    print(x)
# [0 4 8]
# [1 5 9]
# [ 2  6 10]
# [ 3  7 11]

二、多個數組的遍歷

2.1 多個數組Shape相同時

當數組Shape相同時，可以直接使用np.nditer(), 舉例如下：

a = np.arange(12).reshape(3,4)
b = np.ones(12).reshape(3,4)
for i,j in np.nditer([a,b]):
    print(i,"-->",int(j))
# 輸出如下：
# 0 --> 1
# 1 --> 1
# 2 --> 1
# 3 --> 1
# 4 --> 1
# 5 --> 1
# 6 --> 1
# 7 --> 1
# 8 --> 1
# 9 --> 1
# 10 --> 1
# 11 --> 1

2.2 多個數組Shape不同時

當數組Shape不同時，需要滿足廣播規則，才可以使用廣播遍歷，廣播規則請參考：請添加鏈接描述 舉例如下：

a = np.arange(12).reshape(3,4)
b = np.arange(4)
for i,j in np.nditer([a,b]):
    print(i,"-->",j)
# 輸出如下：
# 0 --> 0
# 1 --> 1
# 2 --> 2
# 3 --> 3
# 4 --> 0
# 5 --> 1
# 6 --> 2
# 7 --> 3
# 8 --> 0
# 9 --> 1
# 10 --> 2
# 11 --> 3

感謝各位的閱讀，以上就是“Numpy元素的遍歷方法是什么”的內容了，經過本文的學習后，相信大家對Numpy元素的遍歷方法是什么這一問題有了更深刻的體會，具體使用情況還需要大家實踐驗證。這里是億速云，小編將為大家推送更多相關知識點的文章，歡迎關注！