1. enumerate()函数的用法
enumerate() 函数用于将一个可遍历的数据对象(如列表、元组或字符串)组合为一个索引序列,同时列出数据和数据下标,一般用在 for 循环当中
主要说下当参数为ndarray形式的情况,此时会按照矩阵的第一个维度进行遍历,比如下面的代码:
a = np.array([[1,2],
[3,4],
[5,6]])
for i in enumerate(a):
print(i)
输出的结果是:
(0, array([1, 2]))
(1, array([3, 4]))
(2, array([5, 6]))
可以看到是按照第一个维度进行遍历的
如果进行遍历的时候不仅想要得到值,还想得到对应的索引,则需要用到enumerate() 函数。