Thursday, December 26, 2024
Google search engine
HomeLanguagesRemove multiple elements from a list in Python

Remove multiple elements from a list in Python

Given a list of numbers, write a Python program to remove multiple elements from a list based on the given condition.

Example: 

Input: [12, 15, 3, 10]
Output: Remove = [12, 3], New_List = [15, 10]

Input: [11, 5, 17, 18, 23, 50]
Output: Remove = [1:5], New_list = [11, 50]

Multiple elements can be deleted from a list in Python, based on the knowledge we have about the data. Like, we just know the values to be deleted or also know the indexes of those values. Let’s see different examples based on a different scenarios.

Example #1: Let’s say we want to delete each element in the list which is divisible by 2 or all the even numbers. 

Python3




# Python program to remove multiple
# elements from a list
 
# creating a list
list1 = [11, 5, 17, 18, 23, 50]
 
# Iterate each element in list
# and add them in variable total
for ele in list1:
    if ele % 2 == 0:
        list1.remove(ele)
 
# printing modified list
print("New list after removing all even numbers: ", list1)


Output: 

New list after removing all even numbers:  [11, 5, 17, 23]

Time Complexity: O(n)
Auxiliary Space: O(n)

Example #2: Using list comprehension

Removing all even elements in a list is as good as only including all the elements which are not even( i.e. odd elements). 

Python3




# Python program to remove multiple
# elements from a list
 
# creating a list
list1 = [11, 5, 17, 18, 23, 50]
 
# will create a new list,
# excluding all even numbers
list1 = [ elem for elem in list1 if elem % 2 != 0]
 
print(*list1)


Output: 

11 5 17 23

Time Complexity: O(n)
Auxiliary Space: O(n)

Example #3: Remove adjacent elements using list slicing

Below Python code remove values from index 1 to 4.

Python3




# Python program to remove multiple
# elements from a list
 
# creating a list
list1 = [11, 5, 17, 18, 23, 50]
 
# removes elements from index 1 to 4
# i.e. 5, 17, 18, 23 will be deleted
del list1[1:5]
 
print(*list1)


Output: 

11 50

Time Complexity: O(n)
Auxiliary Space: O(n)

Example #4: Using list comprehension

Let’s say the elements to be deleted is known, instead of the indexes of those elements. In this case, we can directly eliminate those elements without caring about indexes which we will see in next example. 

Python3




# Python program to remove multiple
# elements from a list
 
# creating a list
list1 = [11, 5, 17, 18, 23, 50]
 
# items to be removed
unwanted_num = {11, 5}
 
list1 = [ele for ele in list1 if ele not in unwanted_num]
 
# printing modified list
print("New list after removing unwanted numbers: ", list1)


Output: 

New list after removing unwanted numbers:  [17, 18, 23, 50]

Time Complexity: O(n)
Auxiliary Space: O(n)

Example #5: When index of elements is known.
Though indexes of elements in known, deleting the elements randomly will change the values of indexes. Hence, it is always recommended to delete the largest indices first. Using this strategy, index of smaller values will not be changed. We can sort the list in reverse order and delete the elements of list in descending order.

Python3




# Python program to remove multiple
# elements from a list
 
# creating a list
list1 = [11, 5, 17, 18, 23, 50]
 
# given index of elements
# removes 11, 18, 23
unwanted = [0, 3, 4]
 
for ele in sorted(unwanted, reverse = True):
    del list1[ele]
 
# printing modified list
print (*list1)


Output: 

5 17 50

Method: Using enumerate function

Python3




list1 = [11, 5, 17, 18, 23, 50]
 
list1 = [ elem for i,elem in enumerate(list1) if elem % 2 != 0]
 
print(list1)


Output

[11, 5, 17, 23]

Time Complexity: O(n log n)
Auxiliary Space: O(n)

Approach using numpy.delete():

Algorithm:

Create a numpy array from the given list.
Find the indices of the elements to be removed.
Use numpy.delete() to remove the elements based on the indices found in the previous step.
Convert the resulting numpy array back to a list.
Code:

Python3




import numpy as np
 
#creating a list
list1 = [12, 15, 3, 10]
 
#convert list to numpy array
arr = np.array(list1)
 
#indices of elements to be removed
remove_idx = [0, 2]
 
#use numpy.delete() to remove elements
new_arr = np.delete(arr, remove_idx)
 
#convert numpy array back to list
new_list = new_arr.tolist()
 
#print the results
print("Removed =", [list1[i] for i in remove_idx], ", New_List =", new_list)


Output;

Removed = [12, 3] , New_List = [15, 10]

Time complexity: O(n)
Space complexity: O(n) (for creating the numpy array)

RELATED ARTICLES

Most Popular

Recent Comments