Python function to filter strings with a specific substring

Last update on December 21 2024 07:30:07 (UTC/GMT +8 hours)

Write a Python function that filters out elements from a list of strings containing a specific substring using the filter function.

Sample Solution:

Python Code:

def filter_strings_with_substring(strings, substring):
    """
    Filters out elements from a list of strings containing a specific substring.

    Args:
        strings (list): A list of strings.
        substring (str): The substring to search for in the strings.

    Returns:
        list: A new list containing only the strings that contain the substring.
    """
    # Define the filtering function
    def contains_substring(string):
        return substring in string

    # Use the filter function to filter out strings with the substring
    filtered_strings = list(filter(contains_substring, strings))

    return filtered_strings

# Example usage:
strings = ["Red", "Green", "Orange", "White", "Black", "Pink", "Yellow"]
print("List of words:")
print(strings)
substring = "l"
print("Substring:",substring)
print("Filter out strings with the substring:")
result = filter_strings_with_substring(strings, substring)
print(result)  

Explanation:

In the exercise above -

• First, the "filter_strings_with_substring()" function takes two arguments: strings (a list of strings) and substring (the substring to search for).
• Inside the function, we define a nested function "contains_substring()" that checks if a string contains the specified substring using the in operator.
• Use the filter function to filter out strings from the strings list by applying the "contains_substring()" function as the filtering condition.
• Finally, the filtered strings containing the specified substring are converted to a list and returned as the result.

Sample Output:

List of words:
['Red', 'Green', 'Orange', 'White', 'Black', 'Pink', 'Yellow']
Substring: l
Filter out strings with the substring:
['Black', 'Yellow']

Flowchart:

Python Code Editor:

Previous: Python program to extract words with more than five letters.
Next: Python program to filter future dates.