Python | Remove given element from list of lists
Last Updated :
05 May, 2023
The deletion of elementary elements from list has been dealt with many times, but sometimes rather than having just a one list, we have list of list where we need to perform this particular task. Having shorthands to perform this particular task can help. Let's discuss certain ways to perform this particular task.
Method #1: Using list comprehension The logic behind this kind of method is to reduce the size of code and do the task to perform using loops as a way of list comprehension itself.
Python3
# Python3 code to demonstrate
# Removing element from list of lists
# using list comprehension
# initializing list
test_list = [[4, 5, 6], [5, 6, 4, 1], [4], [4, 8, 9, 10]]
# printing original list
print("The original list : " + str(test_list))
# initializing Number to delete
N = 4
# using list comprehension
# Removing element from list of lists
res = [[ele for ele in sub if ele != N] for sub in test_list]
# print result
print("The list after deletion of element : " + str(res))
Output : The original list : [[4, 5, 6], [5, 6, 4, 1], [4], [4, 8, 9, 10]]
The list after deletion of element : [[5, 6], [5, 6, 1], [], [8, 9, 10]]
Time Complexity: O(n), where n is the number of elements in the list test_list. The list comprehension is used to perform the task and it takes O(n) time.
Auxiliary Space: O(n), additional space of size n is created where n is the number of elements in the new res list.
Method 2: Using list comprehension + list slicing In this method, we generally do the task similar to the above method, the variation is just we use list slicing for better code readability.
Python3
# Python3 code to demonstrate
# Removing element from list of lists
# using list comprehension + list slicing
# initializing list
test_list = [[4, 5, 6], [5, 6, 4, 1], [4], [4, 8, 9, 10]]
# printing original list
print("The original list : " + str(test_list))
# initializing Number to delete
N = 4
# using list comprehension + list slicing
# Removing element from list of lists
for sub in test_list:
sub[:] = [ele for ele in sub if ele != N]
# print result
print("The list after deletion of element : " + str(test_list))
Output : The original list : [[4, 5, 6], [5, 6, 4, 1], [4], [4, 8, 9, 10]]
The list after deletion of element : [[5, 6], [5, 6, 1], [], [8, 9, 10]]
Time complexity: O(n*m), where n is the number of sublists in the list and m is the average length of the sublists.
Auxiliary space: O(1), since the list is modified in place and no additional data structure is used.
Method 3: Using enumerate function
Python3
test_list = [[4, 5, 6], [5, 6, 4, 1], [4], [4, 8, 9, 10]]
N = 4
res = [[ele for j,ele in enumerate(sub) if ele != N] for i,sub in enumerate(test_list)]
print(res)
Output[[5, 6], [5, 6, 1], [], [8, 9, 10]]
Method 4: Using remove() function
This approach involves looping through each sublist and removing the element using the remove() function. It can be implemented as follows:
Python3
def remove_element(lst, element):
# Loop through each sublist
for sub in lst:
# Remove element from sublist
sub.remove(element)
# Initialize list of lists
test_list = [[4, 5, 6], [5, 6, 4, 1], [4], [4, 8, 9, 10]]
# Print original list
print("The original list :", test_list)
# Remove element
remove_element(test_list, 4)
# Print result
print("The list after deletion of element :", test_list)
#This code is contributed by Edula Vinay Kumar Reddy
OutputThe original list : [[4, 5, 6], [5, 6, 4, 1], [4], [4, 8, 9, 10]]
The list after deletion of element : [[5, 6], [5, 6, 1], [], [8, 9, 10]]
The time complexity of the remove_element() function using a for loop is O(n), where n is the total number of elements in the list of lists. This is because the function processes each element in the lists once.
The auxiliary space is O(1), because the function does not create any additional data structures to store the result. It modifies the original list in place.
Method 5: Using recursive function
Using recursive function method, we can remove the element in every nested list
Python3
def remove_element(start,oldlist,newlist,element):
if start==len(oldlist):return newlist #base condition
sublist=[]
for i in oldlist[start]:
if i==element:pass #checking for element
else:sublist.append(i)
newlist.append(sublist) #appending oldlist element to new list
return remove_element(start+1,oldlist,newlist,element)
#Driver code
test_list = [[4, 5, 6], [5, 6, 4, 1], [4], [4, 8, 9, 10]]
print('The Original list: ',test_list)
element=4
print('The List after deleted 4 in every nested list:',remove_element(0,test_list,[],element))
OutputThe Original list: [[4, 5, 6], [5, 6, 4, 1], [4], [4, 8, 9, 10]]
The List after deleted 4 in every nested list: [[5, 6], [5, 6, 1], [], [8, 9, 10]]
Method 6: Using filter()
Python3
test_list = [[4, 5, 6], [5, 6, 4, 1], [4], [4, 8, 9, 10]]
N = 4
print('The Original list: ',test_list)
res = [list(filter(lambda x: x != N, sub)) for sub in test_list]
print("The list after deletion of element : " + str(res))
#This code is contributed by Jyothi pinjala.
OutputThe Original list: [[4, 5, 6], [5, 6, 4, 1], [4], [4, 8, 9, 10]]
The list after deletion of element : [[5, 6], [5, 6, 1], [], [8, 9, 10]]
Time Complexity: O(N*M)
Auxiliary Space: O(N*M)
Method 5: Using map() and filter() functions
This approach involves using the map() and filter() functions in combination with lambda functions to remove the specified element from the list of lists.
Algorithm:
- Initialize a variable "N" to the element that needs to be removed.
- Define a lambda function "remove_element" that returns a filtered sublist by removing the element "N" from the input sublist.
- Use the map() function to apply the "remove_element" function to each sublist in the input list and convert the resulting map object to a list.
Python3
# Python3 code to demonstrate
# Removing element from list of lists
# using map() and filter() functions
# defining function to remove element N
# from a sublist
def remove_element(N, sub): return list(filter((N).__ne__, sub))
# initializing list
test_list = [[4, 5, 6], [5, 6, 4, 1], [4], [4, 8, 9, 10]]
# printing original list
print("The original list : " + str(test_list))
# initializing Number to delete
N = 4
# using map() and filter() functions to
# remove element from list of lists
res = list(map(lambda sub: remove_element(N, sub), test_list))
# print result
print("The list after deletion of element : " + str(res))
OutputThe original list : [[4, 5, 6], [5, 6, 4, 1], [4], [4, 8, 9, 10]]
The list after deletion of element : [[5, 6], [5, 6, 1], [], [8, 9, 10]]
Time Complexity: O(n*m), where n is the number of sublists in the input list and m is the average length of the sublists. This is because the map() function applies the remove_element function to each sublist, which takes O(m) time to execute.
Auxiliary Space: O(n*m), since a new list of lists is created to store the filtered sublists, which has n sublists each of average length m.
Similar Reads
Python | Remove given element from the list Given a list, write a Python program to remove the given element (list may have duplicates) from the given list. There are multiple ways we can do this task in Python. Let's see some of the Pythonic ways to do this task. Example: Input: [1, 8, 4, 9, 2] Output: [1, 8, 4, 2] Explanation: The Element 9
7 min read
Remove Multiple Elements from List in Python In this article, we will explore various methods to remove multiple elements from a list in Python. The simplest way to do this is by using a loop. A simple for loop can also be used to remove multiple elements from a list.Pythona = [10, 20, 30, 40, 50, 60, 70] # Elements to remove remove = [20, 40,
2 min read
Remove last K elements of list - Python Given a list and an integer K, the task is to remove the last K elements from the list. For example, if the list is [1, 2, 3, 4, 5] and K = 2, the result should be [1, 2, 3]. Let’s explore different methods to remove the last K elements from a list in Python.Using list slicingList slicing is one of
3 min read
Remove empty Lists from List - Python In this article, we will explore various method to remove empty lists from a list. The simplest method is by using a for loop.Using for loopIn this method, Iterate through the list and check each item if it is empty or not. If the list is not empty then add it to the result list.Pythona = [[1, 2], [
2 min read
Remove an Element from a List by Index in Python We are given a list and an index value. We have to remove an element from a list that is present at that index value in a list in Python. In this article, we will see how we can remove an element from a list by using the index value in Python. Example: Input: [1, 2, 3, 4, 5], index = 2 Output: [1, 2
3 min read