如何從csv文件讀取2d詞典？

Question

我有一個CSV文件，想讀取該文件以制作2D詞典。

我嘗試創建一個新字典：

f = csv.reader(open('test.csv', 'r'))
for row in f:
    k, v, p  = row 
    markovTransition[k] = {v: p}

上面的代碼提供了我想要的輸出，但當第一個字典的鍵相同時，它將覆蓋鍵。

CSV文件的格式為：

A,A1,3
A,A2,4
B,B1,6
C,C3,7
C,C2,3
C,C5,1

所需的字典是：

{A: {A1: 3, A2: 4}, B: {B1: 6}, C: {C3: 7, C2: 3, C5: 1}

當前的字典是：

{A: {A2: 4}, B: {B1: 6}, C{C5: 1}}

如何從CSV文件創建2d詞典？ 謝謝。

Answer 1

這是defaultdict的一個很好的用例：

markovTransition=collections.defaultdict(dict)
f = csv.reader(open('test.csv', 'r'))
for row in f:
    k, v, p  = row 
    markovTransition[k][v] = p

Answer 2

嘗試這個：

markovTransition = {}
f = csv.reader(open('test.csv', 'r'))
for row in f:
    k, v, p  = row

    if k in markovTransition.keys():  # Check if already exists and then push it.
        markovTransition[k].update({v: p})
    else:
        markovTransition[k] = {v: p}