從數組值中過濾字典

Question

我有一個字典，其值如下：

dictionary = {(10,9): 1, (44,11): 2, (1,1): 99}

基本上我的鍵是整數對，每個鍵的值只是整數。

我有一個存儲一組鍵的數組：

array = [(1,1), (5,19), (58,7)]

我想過濾我的字典，只包含哪些鍵存儲在數組中的元素。 在我的情況下，過濾字典后，我會得到以下內容：

dictionary = {(1,1): 99}

因為存儲在數組中的字典唯一的鍵是(1,1)

最有效的方法是什么？

Answer 1

你可以這樣做：

dictionary = {(10,9): 1, (44,11): 2, (1,1): 99}
array = [(1,1), (5,19), (58,7)]

result = { k:v for k, v in dictionary.items() if k in array}

產量

{(1, 1): 99}

甚至更快，將列表轉換為集合 ：

s = set(array)
result = {k: v for k, v in dictionary.items() if k in s}

Answer 2

您可以找到字典鍵和數組元組的集合交集，然后在字典理解中獲取新值。 這將降低搜索數組中每個鍵的復雜性：

dictionary = {(10,9): 1, (44,11): 2, (1,1): 99}
array = [(1,1), (5,19), (58,7)]

>>> {i:dictionary[i] for i in set(dictionary.keys()).intersection(array)}
{(1, 1): 99}

Answer 3

這是一個詞典理解：

>>> dictionary = {(10,9): 1, (44,11): 2, (1,1): 99}
>>> lst = [(1,1), (5,19), (58,7)]    
>>> d = {k:v for k,v in dictionary.items() if k in lst}
>>> {(1, 1): 99}

我將array重命名為lst因為它是一個list 。 我們不應該將列表與numpy數組， array.array或bytearray類型混淆。

如果你對理解還不熟悉，你也可以寫一個傳統的for循環：

>>> d = {}
>>> for key in dictionary:
...:    if key in lst:
...:        d[key] = dictionary[key]
...:        
>>> d
>>> {(1, 1): 99}

Answer 4

怎么樣：

res = {i: dictionary[i] for i in array if i in dictionary}

這在O（n）中進行 ，其中n是array的項數，不需要任何其他數據結構。

對於你的dictionary和array ，這給了我們：

{(1, 1): 99}