for循环小技巧
- 技巧1:enumerate()
- 技巧2:打包两个可遍历数据,一起循环-zip()
技巧1:enumerate()
在使用pytorch训练网络的过程中,官方教程给出了
for i, data in enumerate(trainloader, 0):
这涉及到enumerate函数的使用,enumerate可以 返回 索引 和 值,同时可以用第二个参数指定索引的初始值(下例子中的1):
list1 = ["这", "是", "一个", "测试"]for index, item in enumerate(list1, 1):print index, item输出:1 这2 是3 一个4 测试
相比与使用 for i in range(len(list))访问列表元素更方便
list1 = ["这", "是", "一个", "测试"]
for i in range (len(list1)):print i ,list1[i]
参考文档:https://blog.csdn.net/hellocsz/article/details/88314997
技巧2:打包两个可遍历数据,一起循环-zip()
两个数据长度一致:
>>> a=[1,2,3,4]
>>> b=[5,6,7,8]
>>> for i,j in zip(a,b):
... print(i,j)
...
1 5
2 6
3 7
4 8
两个数据长度不一致,输出与短数据长度对齐:
>>> a=[1,2,3,4]
>>> c=[0,0]
>>> for i,j in zip(a,c):
... print(i,j)
...
1 0
2 0
>>>