当前位置:   article > 正文

Python的enumerate函数_enumerate(train_loader)

enumerate(train_loader)

Python 的 enumerate() 函数就像是一个神秘的黑箱,你无法简单地用一句话来概括这个函数的作用与用法。

你不再需要在Python代码中专门去生成元素索引,而是将所有这些工作都交给enumerate()函数处理即可。这样,你的代码将更容易被阅读,而且减少写错代码的影响。

另一个有用的特性是,enumerate()函数允许我们为循环自定义起始索引值。enumerate()函数中接受一个可选参数,该参数允许你为本次循环中的计数器变量设置初始值:

  1. names = ["A","B","C","D"]
  2. for index,value in enumerate(names):
  3. print(f'{index}: {value}')
  4. 输出:
  5. 0: A
  6. 1: B
  7. 2: C
  8. 3: D
  9. for index,value in enumerate(names,1):
  10. print(f'{index}: {value}')
  11. 输出:
  12. 1: A
  13. 2: B
  14. 3: C
  15. 4: D

enumerate()函数中加了一个1,输出的索引就是从1开始的。之前默认是0哦。

启发:如果我们需要遍历数据加载器如下:

  1. trainset = namedataset(is_train = True)
  2. trainloader = DataLoader(trainset, batch_size=BATCH_SIZE, shuffle=True)

如果们只需要遍历trainloader的内容:我们只需要 :

  1. for (names, countries) in trainloader:
  2. print(names)
  3. print(countries)

 如果我们需要遍历trainloader的内容并加上索引:我们可以:

  1. for i,(names, countries) in enumerate(trainloader,0):
  2. print(i)
  3. print(names)
  4. print(countries)

如果从0开始enumerate的可选参数就是0,如果从1开始enumerate的可选参数就是1

参考:Python的enumerate函数_Python 学习者的博客-CSDN博客_python enumerate

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小丑西瓜9/article/detail/101810
推荐阅读
相关标签
  

闽ICP备14008679号