python function only returns the first element of the first sublist

Question

I want to print all elements of all sublists which I am able to do with the following loop

sublists=[[1, 2, 3], [4, 5, 6], [7, 7, 7]]
for i in sublists:
    for j in i:
        print(j)

but if I use this function, it only prints the first item of the first sublist. Why is that? How do I make this function work ?

def sl_check(sl):
    for i in sl:
        for j in i:
            return j
print(sl_check(sublists))

Hi, after the return your function doesn't execute anymore, this means only once — baskettaz, Feb 25 '22 at 09:47
If you want to create an iterable object, you can use ```yield``` instead of ```return```. If you want to create a list, use the ```append()``` method — Hobbes, Feb 25 '22 at 09:49
yield returns teh following how to use yield to return the sublists ? — jack johnjacks, Feb 25 '22 at 10:20

score 0 · Answer 1 · answered Feb 25 '22 at 09:49

The return breaks out of the loop, so you'll need to collect all the elements in a new list and then return everything outside the loop:

def sl_check(sl):
    all_elements = []
    for i in sl:
        for j in i:
            all_elements.append(j)
    return all_elements

sublists=[[1, 2, 3], [4, 5, 6], [7, 7, 7]]
print(sl_check(sublists))

python function only returns the first element of the first sublist

1 Answers1

Linked