當鍵是字符串的第二個值時，如何將多個字符串列表附加到字典中的鍵？

Question

假設我有

data = [['Sea', 'Blue', 'Fish', 'Swim'], 
    ['Bored', 'Annoyed', 'Frustrated', 'Done'], 
    ['Keyboard', 'Blue', 'Desktop', 'Mouse']]

我需要具有返回的功能

{'Blue': [['Sea', 'Fish', 'Swim'], ['Keyboard', 'Desktop', 'Mouse']],    
    'Annoyed': [['Bored', 'Frustrated, 'Done']]}

我有這個功能

def func():
    b = dict()
    for element in data:
        for i in element:
                if i[1] in b:
                    b[i[1]].append(i[0], i[2], i[3])
                else:
                    b[i[1]] = (i[0], i[2], i[3])

但它返回：

builtins.IndexError: string index out of range

我希望這是有道理的，預先感謝您

Answer 1

您嵌套了太多的循環。 element已經是內部列表，例如['Sea', 'Blue', 'Fish', 'Swim'等。因此， i是字符串的字符。

還有其他一些問題，請參見下面的評論：

def func():
    b = dict()
    for element in data:
        if element[1] in b:
            b[element[1]].append([element[0], element[2], element[3]]) # append takes one argument
        else:
            b[element[1]] = [[element[0], element[2], element[3]]]  # list, not tuple
    return b