Python – Remove suffix from string list
Last Updated :
28 Feb, 2023
Sometimes, while working with data, we can have a problem in which we need to filter the strings list in such a way that strings ending with specific suffixes are removed. Let’s discuss certain ways in which this task can be performed.
Method #1 : Using loop + remove() + endswith() Method
The combination of the above functions can solve this problem. In this, we remove the elements that end with a particular suffix accessed using a loop and return the modified list.
Python3
test_list = ['allx', 'lovex', 'gfg', 'xit', 'is', 'bestx']
print("The original list : " + str(test_list))
suff = 'x'
for word in test_list[:]:
if word.endswith(suff):
test_list.remove(word)
print("List after removal of suffix elements : "+ str(test_list))
|
Output
The original list : ['allx', 'lovex', 'gfg', 'xit', 'is', 'bestx']
List after removal of suffix elements : ['gfg', 'xit', 'is']
Time Complexity: O(n)
Auxiliary Space: O(1)
Method #2: Using list comprehension + endswith() Method
This is another way in which this task can be performed. In this, we don’t perform the removal in place, instead, we recreate the list without the elements that match the suffix.
Python3
test_list = ['allx', 'lovex', 'gfg', 'xit', 'is', 'bestx']
print("The original list : " + str(test_list))
suff = 'x'
res = [ele for ele in test_list if not ele.endswith(suff)]
print("List after removal of suffix elements : "+ str(res))
|
Output
The original list : ['allx', 'lovex', 'gfg', 'xit', 'is', 'bestx']
List after removal of suffix elements : ['gfg', 'xit', 'is']
Time Complexity: O(n)
Auxiliary Space: O(n)
Method#3: Using filter function + endswith() Method
This is another way in which a task can be performed. In this, we can create a new list with the help of a filter function which filters out all the string which ends with a defined suffix.
Python3
suff = 'x'
def eva(x):
return not x.endswith(suff)
test_list = ['allx', 'lovex', 'gfg', 'xit', 'is', 'bestx']
print("The original list : " + str(test_list))
res = list(filter(eva, test_list))
print("List after removal of Kth character of each string : " + str(res))
|
Output
The original list : ['allx', 'lovex', 'gfg', 'xit', 'is', 'bestx']
List after removal of Kth character of each string : ['gfg', 'xit', 'is']
Time Complexity: O(n)
Auxiliary Space: O(n)
Method #4 : Without any built-in methods
Python3
test_list = ["allx", "lovex", "gfg", "xit", "is", "bestx"]
print("The original list : " + str(test_list))
suff = "x"
res = []
for i in test_list:
if(i[-1] != suff):
res.append(i)
print("List after removal of suffix elements : " + str(res))
|
Output
The original list : ['allx', 'lovex', 'gfg', 'xit', 'is', 'bestx']
List after removal of suffix elements : ['gfg', 'xit', 'is']
Time Complexity: O(n)
Auxiliary Space: O(n)
Method #5 : Using find() and len() methods
Python3
test_list = ["allx", "lovex", "gfg", "xit", "is", "bestx"]
print("The original list : " + str(test_list))
suff = "x"
res = []
for i in test_list:
if(i.find(suff)!=len(i)-1):
res.append(i)
print("List after removal of suffix elements : " + str(res))
|
Output
The original list : ['allx', 'lovex', 'gfg', 'xit', 'is', 'bestx']
List after removal of suffix elements : ['gfg', 'xit', 'is']
Time Complexity: O(n)
Auxiliary Space: O(n)
Method #6 : Using lambda functions
Python3
suff = 'x'
test_list = ['allx', 'lovex', 'gfg', 'xit', 'is', 'bestx']
print("The original list : " + str(test_list))
res = list(filter(lambda x: x[-1] != suff, test_list))
print("List after removal of Kth character of each string : " + str(res))
|
Output
The original list : ['allx', 'lovex', 'gfg', 'xit', 'is', 'bestx']
List after removal of Kth character of each string : ['gfg', 'xit', 'is']
Time Complexity: O(n)
Auxiliary Space: O(n)
Method #7 : Using Regular Expressions
This method uses the regular expression module ‘re’ to match the strings in the list that end with a specific suffix and remove them. Here is an example of how it can be done:
Python3
import re
test_list = ['allx', 'lovex', 'gfg', 'xit', 'is', 'bestx']
print("The original list : " + str(test_list))
suff = 'x$'
res = [x for x in test_list if not re.search(suff, x)]
print("List after removal of suffix elements : " + str(res))
|
Output
The original list : ['allx', 'lovex', 'gfg', 'xit', 'is', 'bestx']
List after removal of suffix elements : ['gfg', 'xit', 'is']
Time Complexity: O(n)
Auxiliary Space: O(n)
Here we use the function re.search() which returns a match object if there is a match anywhere in the string. We use this function in a list comprehension to filter out the elements in the list that match the regular expression.
Method#8: Using pandas
Python3
import pandas as pd
test_list = ['allx', 'lovex', 'gfg', 'xit', 'is', 'bestx']
print("The original list : " + str(test_list))
suffix = 'x'
s = pd.Series(test_list)
res = s[~s.str.endswith(suffix)].tolist()
print("List after removal of suffix elements :", res)
|
Output
The original list : ['allx', 'lovex', 'gfg', 'xit', 'is', 'bestx']
List after removal of suffix elements : ['gfg', 'xit', 'is']
Time Complexity: O(n)
Auxiliary Space: O(n)
Method#8: Recursive method
Algorithm:
- Define a function remove_strings_with_prefix(lst, prefix) that takes in a list lst and a prefix prefix, and returns a new list that includes only those strings in lst that do not end with the prefix.
- Check for the base case where the input list is empty. If the list is empty, return an empty list.
- For the recursive case, remove the prefix from the first element in the list, head, and check whether the resulting string still ends with the prefix.
- If the head element still ends with the prefix, skip it and recurse on the rest of the list by calling remove_strings_with_prefix(lst[1:], prefix).
- If the head element does not end with the prefix, include it in the result by returning [head] + remove_strings_with_prefix(lst[1:], prefix).
- Return the resulting list after all recursive calls have completed.
Python3
def remove_strings_with_prefix(lst, prefix):
if not lst:
return []
head = lst[0]
if head.endswith(prefix):
return remove_strings_with_prefix(lst[1:], prefix)
else:
return [head] + remove_strings_with_prefix(lst[1:], prefix)
test_list = ['allx', 'lovex', 'gfg', 'xit', 'is', 'bestx']
print("The original list : " + str(test_list))
suffix = 'x'
res = remove_strings_with_prefix(test_list,suffix)
print("List after removal of suffix elements :", res)
|
Output
The original list : ['allx', 'lovex', 'gfg', 'xit', 'is', 'bestx']
List after removal of suffix elements : ['gfg', 'xit', 'is']
The time complexity of this method is O(n), where n is the length of the input list, because it processes each element of the list exactly once. The worst-case time complexity occurs when none of the strings in the input list end with the prefix, in which case the method makes n recursive calls. The best-case time complexity occurs when all the strings in the input list end with the prefix, in which case the method makes only one recursive call.
The space complexity of this method is also O(n), because it creates a new list with at most n elements in the worst case. This is because the method creates a new list for each recursive call and concatenates the resulting lists together at each level of recursion. However, in the best case where all the strings in the input list end with the prefix, the method returns the original input list and therefore does not create any new lists.
Please Login to comment...