enumerate 遍歷numpy陣列

覺得有用的話,歡迎一起討論相互學習~Follow Me

遍歷一維陣列

• i，j 分別表示陣列的 索引 和 儲存的值

import numpy as np
a=np.array([1,2,3,4])
for i,j in enumerate(a):
    print("i",i,"j",j)
# i 0 j 1
# i 1 j 2
# i 2 j 3
# i 3 j 4

遍歷二維陣列

• i，j 分別表示陣列的 行的索引 和 每一行中儲存的值

import numpy as np
for i,
 
j in enumerate(b):
    print("i",i,"j",j)
# i 0 j 1
# i 1 j 2
# i 2 j 3
# i 3 j 4
# i 0 j [1 2 3 4]
# i 1 j [5 6 7 8]
# i 2 j [ 9 10 11 12]

遍歷二維陣列集合

for i,(xs,ys) in enumerate(zip(a,b)):
    print("i",i,"(xs,ys)",xs,ys)

# i 0 (xs,ys) 1 [1 2 3 4]
# i 1 (xs,ys) 2 [5 6 7 8]
# i 2 (xs,ys) 3 [ 9 10 11 12]