Get string from list if that string contains substring

Question

Lets say I have following data:

l = ['abc', 'def', 'ghi']
sub = 'bc'

'bc' will allways be a substring for only one element in the list!

and a function that works like this:

def f(l, sub):
    return [x for x in l if sub in x][0]

So the function will return 'abc' because it has 'bc' in it

My Question is: What is the other way to write that function so it doesn't use list index? ([0]) Is there a way to do this any 'prettier'?

If you need only first occurrence, just loop through l and return element when you find it. def f(l, sub): for x in l: if sub in x: return x — Yevhen Kuzmovych
– Yevhen Kuzmovych, Commented Nov 28, 2019 at 14:08

UltraInstinct · Accepted Answer · 2019-11-28 14:08:35Z

8

Use next(..) with a generator. This will sort of break the moment it finds the first instance. Of course, this is assuming one element definitely exists in your list.

def f(l, sub):
    return next(x for x in l if sub in x)

answered Nov 28, 2019 at 14:08

UltraInstinct

44.6k12 gold badges85 silver badges108 bronze badges

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

Comments

Riccardo Bucco · Accepted Answer · 2019-11-28 14:08:27Z

4

Very simple:

next(x for x in l if sub in x)

In this way you don't create the entire list.

answered Nov 28, 2019 at 14:08

Riccardo Bucco

15.5k4 gold badges29 silver badges57 bronze badges

Comments

Raymond Reddington · Accepted Answer · 2019-11-28 14:08:57Z

0

next(x for x in iter(l) if sub in x)

answered Nov 28, 2019 at 14:08

Raymond Reddington

1,8471 gold badge15 silver badges21 bronze badges

Collectives™ on Stack Overflow

Get string from list if that string contains substring

3 Answers 3

Comments

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related