1. 程式人生 > 其它 >在 Python 列表和字典裡,這些坑你踩過嗎

在 Python 列表和字典裡,這些坑你踩過嗎

前言

在Python中,如果我們想要在遍歷一組資料的過程中,對這組資料進行修改,通常會出現許多問題,例如對列表進行上述操作

時, 會忽略部分資料;遍歷字典時,不能修改資料。本文針對這些問題,提出了多種解決方案。

簡介

一、關於列表

1.問題描述

在Python中,如果你試圖在遍歷一組資料的過程中,對其進行修改,這通常沒什麼問題。例如:

Python學習交流Q群:906715085###
l = [3, 4, 56, 7, 10, 9, 6, 5]

for i in l:
    if not i % 2 == 0:
        continue
    l.remove(i)

print(l)

 

上述這段程式碼遍歷了一個包含數字的列表,為了去除掉所有偶數,直接修改了列表l。然而,執行後輸出卻是:

[3, 56, 7, 9, 5]

 

等一下!輸出似乎不對。最終的結果仍然含有一個偶數56。為什麼沒有成功去除這個數呢?我們可以嘗試打印出 for迴圈遍歷的所

有元素,執行如下程式碼:

Python學習交流Q群:906715085###

l = [3, 4, 56, 7, 10, 9, 6, 5]

for i in l:
    print(i)
    if not i % 2 == 0:
        continue
    l.remove(i)

print(l)

 

這段程式碼的輸出為:

3
4
7
10
6
[3, 56, 7, 9, 5]

 

從輸出可以看出,for迴圈似乎沒有訪問列表中的所有元素。為了解for迴圈在內部究竟做了什麼, 我們可以使用 iter 和 next 來模

擬一下。看看下面這個例子,我使用了ipython shell 來執行程式碼:

In [1]: l = [3, 4, 56, 7, 10, 9, 6, 5]

In [2]: # 把列表變成一個迭代器

In [3]: it = iter(l)

In [4]: # 使用 next() 方法來模擬 for迴圈

In [5]: next(it)
Out[
5]: 3 In [6]: next(it) Out[6]: 4 In [7]: # 移除一個迭代器已經訪問過的元素 In [8]: l.remove(3) In [9]: next(it) Out[9]: 7 In [10]: # 注意此處跳過了56,我們可以再移除一個元素 In [11]: l.remove(4) In [12]: next(it) Out[12]: 9

 

上面這個實驗揭示了:當你移除一個迭代器已經訪問過的元素後,在下一次迭代時,會跳過右邊的一個元素,直接訪問下一個。

反之依然成立,即當開始迭代後,如果你在列表開頭添加了一個元素,下次迭代時,可能會訪問到已經迭代過的元素,下面這段

程式碼就出現了這種情況:

In[1]: l = [3, 4, 56, 7, 10, 9, 6, 5]

In[2]: it = iter(l)

In[3]: next(it)
Out[3]: 3

In[4]: next(it)
Out[4]: 4

In[5]: l.insert(0, 44)

In[6]: next(it)
Out[6]: 4

 

注意當在列表頭部添加了44後,4被訪問了兩次。

2.解決方案

為了解決上述問題,我們必須得確保:不能移除迭代器訪問過的元素。

方案一

我們可以先對原列表進行翻轉得到一個新列表,再對新列表進行迭代,並在原列表 l 中移除不符合條件的元素。該方案程式碼如下:

l = [3, 4, 56, 7, 10, 9, 6, 5]

# 迭代翻轉後的列表
for i in reversed(l):
    print(i)
    if not i % 2 == 0:
        continue
    l.remove(i)

print(l)

 

結果如下:

5
6
9
10
7
56
4
3
[3, 7, 9, 5]

 

注意,迭代器現在成功訪問到了列表中的所有元素,並最終輸出了只含有奇數的列表。

方案二

我們還可以在開始迭代前,先複製列表 l 。但是當列表 l 中的資料過多時,這樣做顯然比較耗費效能。該方案程式碼如下:

l = [3, 4, 56, 7, 10, 9, 6, 5]
# 在這裡使用 'l.copy()' 來對列表 l 進行淺拷貝
for i in l.copy():  
    print(i)   
    if not i % 2 == 0:     
        continue  
    l.remove(i)
    
print(l)

 

輸出如下:

3
4
56
7
10
9
6
5
[3, 7, 9, 5]

 

該方案能保證迭代的順序和移除元素的順序相同。不過由於迭代和移除這兩種操作針對的是兩個不同的列表,因此順序相同並不

重要。

二、關於字典

1.問題描述

在對字典進行迭代時,不能修改字典。如下:

# {0: 0, 1: 1, 2: 2, 3: 3, 4: 4, 5: 5, 6: 6, 7: 7, 8: 8, 9: 9}
d = {k: k for k in range(10)}

for k, v in d.items():  
    if not v % 2 == 0:    
        continue  
    d.pop(k)

 

這段程式碼會產生 RuntimeError :

Traceback (most recent call last):  
  File "F:/Documents/pythonprojects/01practice/app.py", line 7, in <module>  
    for k, v in d.items():
RuntimeError: dictionary changed size during iteration

 

2.解決方案

我們可以先複製字典的所有 key ,隨後在迭代 key 的過程中,移除不符合條件的元素。過程如下:

# {0: 0, 1: 1, 2: 2, 3: 3, 4: 4, 5: 5, 6: 6, 7: 7, 8: 8, 9: 9}
d = {k: k for k in range(10)}
# 這裡複製了字典中的所有key值
# 沒有複製整個字典
# 同時使用tuple()速度更快
for k in tuple(d.keys()):   
    if not d[k] % 2 == 0:    
        continue  
    d.pop(k)
    
print(d)

 

執行程式碼後輸出如下:

{1: 1, 3: 3, 5: 5, 7: 7, 9: 9}

 

我們成功移除了字典中的所有偶數鍵值對!

結論

在本文中我們針對迭代一組資料時無法進行修改的問題,分別提出了不同的解決方案:如果想在遍歷列表的時候,對列表進行修

改, 我們可以先對原列表進行翻轉或複製,從而得到一個新列表,隨後在遍歷新列表的過程中,修改原列表中的資料;如果我們

想在遍歷字典的時候,對字典進行修改,可以先複製字典的所有鍵值,然後在迭代鍵值的時候,修改字典中的資料。