我们通常使用一个 for 循环去遍历一个列表,有时候又需要同时遍历多个列表,有什么方法呢?
这个方法就是使用 zip() 函数,里面传入可迭代对象。
zip(列表 1,列表 2,…) 将多个列表对应位置的元素组合成为元组,并返回这个 zip 对象。
示例代码:
list1 = [1, 2, 3, 4]
list2 = [9, 8, 7, 6]
z = zip(list1, list2)
print(z)
for i, j in z:
print(i + j)
# 运行结果:
<zip object at 0x00000159B3B5C348>
10
10
10
10
我们继续深究一下,这个返回的 zip 对象是一个什么类型的对象呢?
我们在上面的代码中增加一句 print(list(z))
,将 返回的 zip 对象转换成一个列表:
list1 = [1, 2, 3, 4]
list2 = [9, 8, 7, 6]
z = zip(list1, list2)
print(z)
print(list(z))
for i, j in z:
print(i + j)
其运行结果如下:
<zip object at 0x00000159B3B5C348>
[(1, 9), (2, 8), (3, 7), (4, 6)]
Process finished with exit code 0
发现后面的 for 循环并没有输出结果,只输出了一行空白,这又是什么原因呢?发现跟遍历迭代器的很像。
原来 zip 对象是一个迭代器类型的对象,迭代器在第一次时已经遍历结束了,第二次不能再使用。
可再次通过 next() 或者 __next__() 检验其是否为迭代器:
list1 = [1, 2, 3, 4]
list2 = [9, 8, 7, 6]
z = zip(list1, list2)
print(z.__next__())
print(z.__next__())
print(z.__next__())
print(z.__next__())
print(z.__next__())
运行结果为:
(1, 9)
(2, 8)
(3, 7)
(4, 6)
Traceback (most recent call last):
File "E:/PythonProject/test.py", line 707, in <module>
print(z.__next__())
StopIteration
可知 zip() 函数返回的结果就是一个迭代器。