首頁 > 軟體

Python中的 enumerate和zip詳情

2022-05-30 22:02:09

前言

我們在上一期學習了關於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內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com!


IT145.com E-mail:sddin#qq.com