Python中的 enumerate和zip

Posted on by

前言

我們在上一期學習瞭關於Python 迭代器Iterator詳情相關的概念，滿足迭代器需要符合兩個條件

  • 實現__iter__()方法：返回迭代器本省
  • 實現__next__()方法：返回迭代器的下一項

同時，Python中提供數據類型如列表、字典、元組等序列是可迭代對象，可結合for…in 來進行遍歷為使我們的代碼更加高效且優雅，Python也提供兩個關於迭代的方法。

因此，本期我們來學習enumerate函數和zip函數的相關使用方法，Let's go~~

1. enumerate 方法

我們在上述瞭解瞭迭代器的相關的知識點，對於可迭代對象列表、字典等可以使用for…in來進行快速遍歷。例如對序列items使用for循環進行如下遍歷，不用進行計算長度多餘步驟。

for item in items:

但是，如上的遍歷無法知道items序列每一個元素的索引，如果實際場景中，需要使用元素的索引呢？通常，我們會創建一個變量index，來記錄元素的索引位置

index = 0
for item in items:
    # options
    index += 1

雖然上述方式可以解決我們的需求，但是代碼顯示非常的冗餘，不符合pythonic的規范

在Pythonic要求中，我們可以使用enumerate()函數來替代上述實現方法

for index,item in enumerate(items):
    # options
    ...
  • enumerate()函數返回是一個enumerate對象實例，它是一個迭代器。
  • enumerate對象返回是連續包含一個由計數和值組成的元組(index,value)
  • 元組中value值通過傳入序列上調用的**next()**返回的

但是，當序列中的元素是元組類型時，在使用enumerate()函數時，需要對value進行嚴格區分。

data = [(1,2),(3,4),(5,6),(7,8)]
# correct
for index,(x,y) in enumerate(data):
    ...
# Error
for index,x,y in enumerate(data):
    ...

2. zip 方法

enumerate()函數時針對單個序列遍歷處理。zip則是針對需要處理兩個及以上的序列遍歷處理。

zip 處理的是多對象迭代的，比如要同時迭代兩列表name,age。按照傳統思路隻能使用下標遍歷的方式，對於迭代器則運用不上。

但是使用下標遍歷的方式，會有兩個問題：

  • 代碼冗餘，可讀性差，並且不夠PythonIC
  • 如果兩個容器的列表長度不一樣時，則會需要增加額外代碼來維護

Python 提供zip()函數可以解決以上兩問題。

names = ["bone","anne","tom","tony"]
ages = [12,16,19]
for name,age in zip(names,ages):
    print(name,age)
...
bone 12
anne 16
tom 19
...

zip(x,y)會自動返回一個元組(a,b)的迭代器，其中x來自a,y來自b。當某個序列到結尾瞭，迭代也會隨之結束。迭代長度跟參數中最短序列長度一致。

names = ["bone","anne","tom","tony"]
ages = [12,16,19]
for stu in zip(names,ages):
    print(stu)
...
('bone', 12)
('anne', 16)
('tom', 19)
...

當然，如果要以長度最長的列表為一致時，我們可以選擇 iterators.zip_longset() 函數來代替。

from itertools import zip_longest
for stu in zip_longest(names,ages):
    print(stu)
...
('bone', 12)
('anne', 16)
('tom', 19)
('tony', None)
...

同時，使用zip()函數可以方便對兩列表打包生成字典：

s = dict(zip(names,ages))
print(s)
...
{'bone': 12, 'anne': 16, 'tom': 19}
...

zip()會創建一個迭代器來作為結果返回的。如果將結果需要存儲在列表中，則需要list()進行轉換

print(list(zip(names,ages)))

...
[('bone', 12), ('anne', 16), ('tom', 19)]
...

總結

本期對Python針對序列遍歷符合PythonIC的兩種方法 enumerate函數和zip函數enumerate函數可以方便處理索引問題，zip函數處理多對象迭代場景。

到此這篇關於Python中的 enumerate和zip的文章就介紹到這瞭,更多相關Python enumerate和zip內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：