1. 程式人生 > 其它 >Python程式碼閱讀(第4篇):過濾掉列表中的唯一值

Python程式碼閱讀(第4篇):過濾掉列表中的唯一值

本篇閱讀的程式碼實現了過濾掉列表中的唯一值的功能。 本篇閱讀的程式碼片段來自於30-seconds-of-python。

本篇閱讀的程式碼實現了過濾掉列表中的唯一值的功能。

本篇閱讀的程式碼片段來自於30-seconds-of-python。

filter_unique

from collections import Counter

def filter_unique(lst):
  return [item for item, count in Counter(lst).items() if count > 1]

# EXAMPLES
filter_unique([1, 2, 2, 3, 4, 4, 5]) # [2, 4]

函式使用collections.Counter函式對列表進行計數,在通過列表推導式,將非唯一值過濾出去(過濾掉計數大於1的值)。

collections.Counter

class collections.Counter([iterable-or-mapping])

Counter是一個dict的子類,用於計數可雜湊物件。它是一個集合,元素像字典鍵(key)一樣儲存,它們的計數儲存為值。計數可以是任何整數值,包括0和負數。它可以接收一個可迭代物件,並對其元素進行計數。

其成員函式items()將返回一個字典的鍵值對的新檢視。再結合collections.Counter,上述程式碼中Counter(lst).items()將一個列表轉化成了(元素,計數)對。

列表推導式

列表推導式提供了一個更簡單的建立列表的方法。常見的用法是把某種操作應用於序列或可迭代物件的每個元素上,然後使用其結果來建立列表,或者通過滿足某些特定條件元素來建立子序列。

列表推導式的結構是由一對方括號所包含的以下內容:一個表示式,後面跟一個for子句,然後是零個或多個forif子句。其結果將是一個新列表,由對錶達式依據後面的forif子句的內容進行求值計算而得出。

上面filter_unique可以等價的寫成:

from collections import Counter

def filter_unique(lst):
  temp_list = []
  for item, count in Counter(lst).items():
    if count > 1:
      temp_list.append(item)
  return temp_list

# EXAMPLES
filter_unique([1, 2, 2, 3, 4, 4, 5]) # [2, 4]

filter_non_unique

from collections import Counter

def filter_non_unique(lst):
  return [item for item, count in Counter(lst).items() if count == 1]

# EXAMPLES
filter_non_unique([1, 2, 2, 3, 4, 4, 5]) # [1, 3, 5]

filter_non_unique和上面的程式碼相似,功能是過濾掉列表中的非唯一值。