Python：將與鍵相關聯的列表值存儲在字典中

Question

我知道python字典如何存儲鍵：值元組。 在我正在進行的項目中，我需要存儲與列表值關聯的鍵。 例如：key-> [0,2,4,5,8]其中，key是文本文件中的單詞，列表值包含表示該單詞所在的DocID的整數。

一旦我在另一個文檔中找到相同的單詞，就需要將該DocID附加到列表中。

我該如何實現？

Answer 1

您可以使用默認值，如下所示：

>>> import collections
>>> d = collections.defaultdict(list)
>>> d['foo'].append(9)
>>> d
defaultdict(<type 'list'>, {'foo': [9]})
>>> d['foo'].append(90)
>>> d
defaultdict(<type 'list'>, {'foo': [9, 90]})
>>> d['bar'].append(5)
>>> d
defaultdict(<type 'list'>, {'foo': [9, 90], 'bar': [5]})

Answer 2

這篇文章對我解決在動態創建帶有附加數據列表的變量鍵時遇到的問題很有幫助。 見下文：

import collections

d = collections.defaultdict(list)
b = collections.defaultdict(list)
data_tables = ['nodule_data_4mm_or_less_counts','nodule_data_4to6mm_counts','nodule_data_6to8mm_counts','nodule_data_8mm_or_greater_counts']

for i in data_tables:
    data_graph = con.execute("""SELECT ACC_Count, COUNT(Accession) AS count
                                            FROM %s
                                            GROUP BY ACC_Count"""%i)
    rows = data_graph.fetchall()
    for row in rows:
        d[i].append(row[0])
        b[i].append(row[1])

print d['nodule_data_4mm_or_less_counts']
print b['nodule_data_4mm_or_less_counts']

它輸出每個鍵的數據列表，然后可以更改為np.array進行繪圖等。

>>>[4201, 1052, 418, 196, 108, 46, 23, 12, 11, 8, 7, 2, 1]
>>>[ 0  1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16]

Answer 3

這將是使用defaultdict的好地方

from collections import defaultdict

docWords = defaultdict(set)
for docID in allTheDocIDs:
    for word in wordsOfDoc(docID):
        docWords[word].add(docID)

您可以使用列表而不是集合

Answer 4

像這樣嗎

word = 'something'
l = [0,2,4,5,8]
myDict = {}
myDict[word] = l

#Parse some more

myDict[word].append(DocID)

Answer 5

我曾經寫過一個幫助器類，以使@Vinko Vrsalovic的答案更易於使用：

class listdict(defaultdict):
    def __init__(self):
        defaultdict.__init__(self, list)

    def update(self, E=None, **F):
        if not E is None:
            try:
                for k in E.keys():
                    self[k].append(E[k])
            except AttributeError:
                for (k, v) in E:
                    self[k].append(v)
        for k in F:
            self[k].append(F[k])

可以這樣使用：

>>> foo = listdict()
>>> foo[1]
[]
>>> foo.update([(1, "a"), (1, "b"), (2, "a")])
>>> foo
defaultdict(<type 'list'>, {1: ['a', 'b'], 2: ['a']})

Answer 6

如果我的問題正確，您可以嘗試一下，

           >>> a=({'a':1,'b':2});
           >>> print a['a']
            1
           >>> a.update({'a':3})
           >>> print a['a']
            3
            >>> a.update({'c':4})
            >>> print a['c']
             4

這將適用於舊版本的python