enumerate 遍历numpy数组

演化计算与人工智能

发布于 2020-08-14 00:15:37

1.5K00

代码可运行

文章被收录于专栏：人工智能与演化计算成长与进阶人工智能与演化计算成长与进阶

运行总次数：0

代码可运行

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]

本文参与腾讯云自媒体同步曝光计划，分享自微信公众号。

原始发表：2020-06-14，如有侵权请联系 cloudcommunity@tencent.com 删除

编程算法