enumerate 遍历numpy数组

时间:2022-07-23
本文章向大家介绍enumerate 遍历numpy数组,主要内容包括其使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。

enumerate 遍历numpy数组

遍历一维数组

  • i,j 分别表示数组的 索引存储的值
import numpy as np
a=np.array([1,2,3,4])
b=np.array([[1,2,3,4],[5,6,7,8],[9,10,11,12]])
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]