How to return elements using for loop within a function?

Question

Below is the given list of strings from which I am trying to find the strings which contains the substring. Output is obtained perfectly without using functions. However when function is used, it does return only 1 substring and not the list of strings containing the substring.

L = ['dog', 'dodge', 'cat', 'cattle']

sub_string = str(input("Enter the substring: "))

for i in range(len(L)):
    v = L[i].find(sub_string, 0,100)
    if v >= 0:
        print(L[i])           #returns all the strings containing the substring very well.

def String_find(sub_string):
    for i in range(len(L)):
        if sub_string in L[i]:         
            return L[i]        #returns only 'dog' or 'cat' if substring is entered whereas 
                               #'dog','dodge' 
                               #or 'cat','cattle' is expected.

Total = String_find(sub_string)
#Output is obtained perfectly without using functions however when function is used it does return         
#only 1 substring and not the list of strings containing the substring.

return returns immediately, so the first match will trigger the exit of the function. You may be lookiing for yield, or to return a list that you fill with matches. — L3viathan
– L3viathan, Commented Mar 27, 2020 at 15:43
return will cause the loop to exit. So as soon as a match is found, it exits from the loop. — dEBA M
– dEBA M, Commented Mar 27, 2020 at 15:47

kyore · Accepted Answer · 2020-03-27 15:47:46Z

3

You maybe want to do this:

def string_find(sub):
    return [item for item in L if sub in item]

items = string_find(sub_string)
print(items)

answered Mar 27, 2020 at 15:47

kyore

8425 silver badges25 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Aztec-3x5 Over a year ago

Most optimal solution

Collectives™ on Stack Overflow

How to return elements using for loop within a function?

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related