当键是字符串的第二个值时，如何将多个字符串列表附加到字典中的键？

Question

Let's say I have

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

I need to have the function to return

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

I have this function I made

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])

But it returns:

builtins.IndexError: string index out of range

I hope this made sense, thank you in advance

Answer 1

You are nesting too many loops. element is already the inner list like ['Sea', 'Blue', 'Fish', 'Swim' , etc. i is therefore the characters of the string.

There are some other problems, see the comments below:

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