Python: Find the strings in a list containing a given substring

Python Programming Puzzles: Exercise-16 with Solution

Write a Python program find the strings in a given list containing a given substring.

Input:
[(ca,('cat', 'car', 'fear', 'center'))]
Output:
['cat', 'car']

Input:
[(o,('cat', 'dog', 'shatter', 'donut', 'at', 'todo', ''))]
Output:
['dog', 'donut', 'todo']

Input:
[(oe,('cat', 'dog', 'shatter', 'donut', 'at', 'todo', ''))]
Output:
[]

Pictorial Presentation:

Sample Solution:

Python Code:

def test(strs, substr):
    return [s for s in strs if substr in s]
strs =  ['cat', 'car', 'fear', 'center']
print("Original strings:")
print(strs)
substrs = 'ca'
print("Substring: "+substrs)
print("Strings in the said list containing a given substring:")
print(test(strs, substrs))
strs =  ['cat', 'dog', 'shatter', 'donut', 'at', 'todo', '']
print("\nOriginal strings:")
print(strs)
substrs = 'o'
print("Substring: "+substrs)
print("Strings in the said list containing a given substring:")
print(test(strs, substrs))
strs =  ['cat', 'dog', 'shatter', 'donut', 'at', 'todo', '']
print("\nOriginal strings:")
print(strs)
substrs = 'oe'
print("Substring: "+substrs)
print("Strings in the said list containing a given substring:")
print(test(strs, substrs))

Sample Output:

Original strings:
['cat', 'car', 'fear', 'center']
Substring: ca
Strings in the said list containing a given substring:
['cat', 'car']

Original strings:
['cat', 'dog', 'shatter', 'donut', 'at', 'todo', '']
Substring: o
Strings in the said list containing a given substring:
['dog', 'donut', 'todo']

Original strings:
['cat', 'dog', 'shatter', 'donut', 'at', 'todo', '']
Substring: oe
Strings in the said list containing a given substring:
[]

Flowchart:

Visualize Python code execution:

The following tool visualize what the computer is doing step-by-step as it executes the said program:

Python Code Editor :

Have another way to solve this solution? Contribute your code (and comments) through Disqus.

Previous: Find the longest string of a list of strings.
Next: Find a string consisting of the non-negative integers up to n inclusive.