0
python 就像一件藝術(shù)珍藏品!
python 最大的優(yōu)點之一是它可以廣泛地選擇模塊和包。它們將 python 的功能擴展到許多流行的領(lǐng)域,包括機器學(xué)習(xí)、數(shù)據(jù)科學(xué)、web 開發(fā)、前端等等。其中最好的一個優(yōu)點是 python 的內(nèi)置 collections 模塊。
在一般意義上,python 中的集合是用于存儲數(shù)據(jù)集合(如 list、dict、tuple 和 set)的容器。這些容器直接構(gòu)建在 python 中,可以直接調(diào)用。collections 模塊提供額外的高性能數(shù)據(jù)類型,這些數(shù)據(jù)類型可以提高代碼的性能。
讓我們?yōu)g覽一下 collections 模塊最流行的數(shù)據(jù)類型以及如何使用它們吧!
1.counter
counter 是 dictionary 對象的一個子類。collections 模塊中的 counter()函數(shù)接受 iterable,例如 list 或 tuple,并返回計數(shù)器字典。字典的鍵將是 iterable 的唯一元素,每個鍵的值將是 iterable 中元素的計數(shù)。
作為開始,讓我們從集合導(dǎo)入計數(shù)器數(shù)據(jù)類型:
from collections import Counter
若要創(chuàng)建計數(shù)器對象,請將其分配給變量,這和任何其他對象類是一樣的。你唯一想傳給它的參數(shù)就是你的變量。
lst = [1, 2, 3, 3, 2, 1, 1, 1, 2, 2, 3, 1, 2, 1, 1]
counter = Counter(lst)
如果我們在對象 print(counter)周圍使用一個簡單的 print 函數(shù)來打印計數(shù)器,我們會得到一個看起來有點像字典的東西:
Counter({1: 7, 2: 5, 3: 3})
你可以使用其 key 訪問任何計數(shù)器項,如下所示。這與從標(biāo)準(zhǔn) python 字典中提取元素的方法完全相同。
lst = [1, 2, 3, 3, 2, 1, 1, 1, 2, 2, 3, 1, 2, 1, 1]
counter = Counter(lst)
print(counter[1])
most_common() 函數(shù)
到目前為止,計數(shù)器對象最有用的函數(shù)是 most_common() 函數(shù)。當(dāng)它應(yīng)用于計數(shù)器對象時,它返回 n 個最常見元素及其計數(shù)的列表,按從最常見到最少見的順序排列。
lst = [1, 2, 3, 3, 2, 1, 1, 1, 2, 2, 3, 1, 2, 1, 1]
counter = Counter(lst)
print(counter.most_common(2))
上面的代碼打印出以下元組列表:
[(1,7),(2,5)]
每個元組的第一個元素是列表中唯一的項,每個元組的第二個元素是計數(shù)。這是一種簡單快捷的方法,比如「獲取列表中最常見的前 3 個元素及其計數(shù)」。
要了解更多有關(guān)計數(shù)器功能的信息,請查看官方文檔。
2.defaultdict
該函數(shù)的工作原理與普通的 python 字典完全相同,額外的好處是當(dāng)你試圖訪問一個不存在的鍵時,它不會拋出錯誤。
相反,它使用默認值初始化 key。默認值是根據(jù)創(chuàng)建 DefaultDict 對象時作為參數(shù)傳遞的數(shù)據(jù)類型自動設(shè)置的。以下面的代碼為例。
from collections import defaultdict
names_dict = defaultdict(int)
names_dict["Bob"] = 1
names_dict["Katie"] = 2
sara_number = names_dict["Sara"]
print(names_dict)
在上面的示例中,int 作為默認值傳遞給 defaultdict 對象。接下來,為每個鍵定義值,為鍵「bob」和「katie」定義數(shù)值。但在最后一行,我們試圖訪問一個尚未定義的 key,即「sara」。
在普通字典中,這會拋出一個錯誤。但是使用 defaultdict,一個新的 key 會自動初始化為「sara」,值為 0,對應(yīng)于我們的 int 數(shù)據(jù)類型。因此,最后一行打印出一個包含所有 3 個名稱和相應(yīng)值的字典。
defaultdict(<class'int'>,{'bob':1,'katie':2,'sara':0})
如果我們用一個類似于這個名稱的列表初始化 defaultdict,那么「sara」將用一個空列表 [] 初始化,代碼將輸出以下內(nèi)容:
defaultdict(<class'int'>,{'bob':1,'katie':2,'sara':[]})
要了解更多有關(guān) DefaultDict 功能的信息,請查看官方文檔。
3.deque
隊列是計算機科學(xué)中遵循先進先出(fifo)原則的基本數(shù)據(jù)結(jié)構(gòu)。簡單地說,這意味著添加到隊列中的第一個對象也必須是要刪除的第一個對象。我們只能在隊列前面插入內(nèi)容,只能從后面刪除內(nèi)容——隊列中間不能執(zhí)行任何操作。
collections 庫的 deque 實現(xiàn)了該功能的優(yōu)化版本。該實現(xiàn)的一個關(guān)鍵特性是保持隊列大小,即如果將隊列的最大大小設(shè)置為 10,則 deque 將根據(jù) fifo 原則添加和刪除元素,以保持最大大小為 10。這是目前為止 python 中隊列的最佳實現(xiàn)。
讓我們從下面這個例子開始吧。我們要創(chuàng)建一個 deque 對象,然后用從 1 到 10 的整數(shù)初始化它。
from collections import deque
my_queue = deque(maxlen=10)
for i in range(10):
my_queue.append(i+1)
print(my_queue)
在上面的代碼中,我們首先初始化了我們的 deque,指定我們希望它始終保持最大的長度 10。其次,我們通過循環(huán)將值插入到隊列中。請注意,填充隊列的功能與使用常規(guī) python 列表完全相同。最后,我們打印出結(jié)果。
deque([1, 2, 3, 4, 5, 6, 7, 8, 9, 10], maxlen=10)
因為我們的隊列有一個 maxlen=10,并且我們的循環(huán)添加了 10 個元素,所以我們的隊列包含從 1 到 10 的所有數(shù)字?,F(xiàn)在讓我們看看當(dāng)我們增加更多的數(shù)字時會發(fā)生什么。
for i in range(10, 15):
my_queue.append(i+1)
print(my_queue)
上面的代碼中,我們向隊列中添加了另外 5 個元素,即從 11 到 15 的數(shù)字。但是我們的隊列只有 maxlen=10,所以它必須刪除一些元素。由于隊列必須遵循 fifo 原則,因此它會刪除插入隊列的前 5 個元素,正是按照它們的插入順序:[1、2、3、4、5]。print 語句的結(jié)果如下:
deque([6, 7, 8, 9, 10, 11, 12, 13, 14, 15], maxlen=10)
要了解更多有關(guān) deque 的信息,請查看官方文檔。
4.namedtuple
在 python 中創(chuàng)建常規(guī)元組時,其元素是通用的和未命名的。這迫使你記住每個元組元素的確切索引。namedtuple 就是這個問題的解決方案。
namedtuple()返回一個元組,該元組中每個位置的名稱都是固定的,而 namedtuple 對象的名稱是通用的。要使用 namedtuple,首先為它創(chuàng)建一個模板。下面的代碼創(chuàng)建一個名為「person」的 namedtuple 模板,該模板具有「name」、「age」和「job」參數(shù)。
from collections import namedtuple
Person = namedtuple('Person', 'name age job')
創(chuàng)建模板后,可以使用它創(chuàng)建 namedtuple 對象。讓我們?yōu)?2 個 person 創(chuàng)建 2 個 namedtuple 并打印出他們的表示。
Person = namedtuple('Person', 'name age job')
Mike = Person(name='Mike', age=30, job='Data Scientist')
Kate = Person(name="Kate", age=28, job='Project Manager')
print(Mike)
print(Kate)
上面的代碼非常直截了當(dāng)——我們用 namedtuple 模板的所有屬性初始化一個「person」。上面的打印語句將給出以下結(jié)果:
Person(name='Mike', age=30, job='Data Scientist')
Person(name='Kate', age=28, job='Project Manager')
因此,namedtuples 允許更大的易用性、可讀性和更容易組織元組對象。
要了解 namedtuple 的更多功能,請查看官方文檔。
結(jié)論
好了,你學(xué)完這些啦!接下來你可以使用 collections 庫使用 python 中的高性能數(shù)據(jù)類型了~
如果你渴望更多,別擔(dān)心!在 python 集合中還有很多東西需要學(xué)習(xí),你還需要學(xué)習(xí)如何最有效地使用它們。作為推薦閱讀材料,pymotw 網(wǎng)站是一個很好的選擇。如果你想了解更多關(guān)于 python 的基礎(chǔ)知識,coursera 有一個“Python for Everybody”課程,該課程更側(cè)重于應(yīng)用程序,快去加入學(xué)習(xí)吧~
雷鋒網(wǎng)雷鋒網(wǎng)雷鋒網(wǎng)
雷峰網(wǎng)版權(quán)文章,未經(jīng)授權(quán)禁止轉(zhuǎn)載。詳情見轉(zhuǎn)載須知。