Python：多維字典

Question

我運行了幾次具有不同輸入參數的應用程序，以收集執行時間。

的輸入參數爲6：v，n，m，b，p和c。

概念我覺得我的結果多維數組，其中的任何尺寸是不同的參數：times[A][B][C][D][E][F]將包含使用參數v=A，n=B，m=C，b=D，p=E和c=F仿真的執行時間。

我希望能夠修復一些參數和迭代其他：

for A: 
    for C: 
    for F: 
     times[A][0][C][0][0][F] 

輸入參數值是稀疏的，所以我應該使用字典而不是列表。

我想使用字典做整個事情的字典的字典的字典的字典的字典，每個執行時間將這樣表示：

times = { A:{ B:{ C:{ D:{ E:{ F:{time} } } } } } } 

但這種方法不看優雅：構建整個結構並迭代它是一種痛苦。

有沒有更好的方式來處理我的數據？

Answer 1

首先，如果你必須使用字典，爲什麼不創建一個單一的字典，使用元組索引它呢？二，使用itertools.product避免麻煩的嵌套循環：

>>> import itertools 
>>> d = {} 
>>> for tup in itertools.product(range(5), repeat=2): 
...  d[tup] = tup 
... 
>>> d 
{(1, 3): (1, 3), (3, 0): (3, 0), (2, 1): (2, 1), (0, 3): (0, 3), (4, 0): (4, 0), 
(1, 2): (1, 2), (3, 3): (3, 3), (4, 4): (4, 4), (2, 2): (2, 2), (4, 1): (4, 1), 
(1, 1): (1, 1), (3, 2): (3, 2), (0, 0): (0, 0), (0, 4): (0, 4), (1, 4): (1, 4), 
(2, 3): (2, 3), (4, 2): (4, 2), (1, 0): (1, 0), (0, 1): (0, 1), (3, 1): (3, 1), 
(2, 4): (2, 4), (2, 0): (2, 0), (4, 3): (4, 3), (3, 4): (3, 4), (0, 2): (0, 2)} 

然而，可能會有更好的方式來創建一個稀疏數組。 scipy提供sparse matrices，但我相信它們只是2-d。

這裏有一些其他的使用模式可能對您有用：

>>> for tup in itertools.product(range(5), repeat=2): 
...  if tup[0] == tup[1]: 
...   d[tup] = tup 
... 
>>> d 
{(3, 3): (3, 3), (0, 0): (0, 0), (1, 1): (1, 1), (4, 4): (4, 4), (2, 2): (2, 2)} 

>>> for tup in itertools.product(range(5), range(2)): 
...  print d.get(tup) 
... 
(0, 0) 
None 
None 
(1, 1) 
None 
None 
None 
None 
None 
None 

要斜少，這裏是你將如何保持一個變量不變：只是傳遞一個項目順序itertools.product：

>>> for tup in itertools.product(range(3), [2], range(3)): 
...  print tup 
... 
(0, 2, 0) 
(0, 2, 1) 
(0, 2, 2) 
(1, 2, 0) 
(1, 2, 1) 
(1, 2, 2) 
(2, 2, 0) 
(2, 2, 1) 
(2, 2, 2) 

Answer 2

您可以使用元組內的參數組合作爲關鍵字創建一維字典，這會更簡單：

times[('v', 'n', 'm', 'b', 'p', 'c')] = value