how to fetch whole string if part of the sting is present in list using python

Swati B 0
Viewed 33

I want to search the following list, to see if any of the string items contain 'aed'; if so, I want to return the whole string (e.g. 'abc/aed'). How can this be achieved using python/pyspark?

list=['abc/aed','bcd/eac','rtf/reew','opee/rew']
pyspark python-3.x python
4 Answers

this should work

my_list=['abc/aed','bcd/eac','rtf/reew','opee/rew']
output = [val for val in my_list if "aed" in val]
Lucas M. Uriarte0

We can use a list comprehension:

list = ['abc/aed', 'bcd/eac', 'rtf/reew', 'opee/rew']
output = [x for x in list if "aed" in x]
print(output)  # ['abc/aed']
Tim Biegeleisen0 
for i in list:
    if "aed" in i:
        print(i)
Kra1ven0

Use a map for python list. e.g.,

>>> l = ['abc/aed','bcd/eac','rtf/reew','opee/rew']
>>> list(map(lambda x: x if x.find("aed") != -1 else False, l))
['abc/aed', False, False, False]
the_ordinary_guy0
Related