我的循环在第一次迭代后停止

Question

I am trying to loop through a dictionary of lists and return the keys of values which contain a specific integer. For example if I loop through {0: [3], 1: [3], 2: [4, 0], 3: [1], 4: [1, 0, 2, 3]} with v = 4, it should return [2]. However, my code seems to only consider the very first key value pair and I do not understand why. It works if the integer is in the first key value pair and not in any others. Here is the function I made:

def whence(g, v):
    # Your code here
    lov = []
    count = 0 
    for key, value in g.items(): 
        if v in value:
            lov.append(count)
        count += 1
        print(lov)
        return lov

Answer 1

You can do it in one line:

def whence(g, v):
    return [key for key, values in g.items() if v in values]

Answer 2

The return statement indented too much.

def whence(g, v):
    # Your code here
    lov = []
    count = 0 
    for key, value in g.items(): 
        if v in value:
            lov.append(count)
        count += 1
        print(lov)
    return lov