Python - Remove given element from list of lists

Python lists are versatile data structures that can contain other lists, creating nested or 2D structures. When working with lists of lists, you may need to remove specific elements from the inner lists. Python provides several approaches to accomplish this task efficiently.

Understanding Lists of Lists

A list of lists is a collection where each element is itself a list. This creates a matrix-like structure that's useful for storing tabular data or multi-dimensional information.

# Example of a list of lists
matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
print("Original matrix:", matrix)

Original matrix: [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

Using Nested Loop

The most straightforward approach uses a nested loop to iterate through each sublist and remove the target element ?

# Sample list of lists
matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
element_to_remove = 5

# Iterate over each sublist
for sublist in matrix:
    if element_to_remove in sublist:
        sublist.remove(element_to_remove)

print("After removing", element_to_remove, ":", matrix)

After removing 5 : [[1, 2, 3], [4, 6], [7, 8, 9]]

How It Works

This method iterates through each sublist, checks if the target element exists using the in operator, and removes it using the remove() method. The remove() method only removes the first occurrence of the element in each sublist.

Using List Comprehension

List comprehension provides a more concise and Pythonic solution by creating a new list structure ?

# Sample list of lists
data = [[1, 2, 3], [4, 5, 6], [7, 2, 9]]
element_to_remove = 2

# Using list comprehension to remove element
result = [[item for item in sublist if item != element_to_remove] for sublist in data]

print("Original:", data)
print("After removing", element_to_remove, ":", result)

Original: [[1, 2, 3], [4, 5, 6], [7, 2, 9]]
After removing 2 : [[1, 3], [4, 5, 6], [7, 9]]

Key Difference

Unlike the nested loop approach, list comprehension removes all occurrences of the element from each sublist, not just the first one.

Removing Multiple Elements

You can extend these approaches to remove multiple elements at once ?

# Remove multiple elements using list comprehension
data = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
elements_to_remove = [2, 5, 8]

result = [[item for item in sublist if item not in elements_to_remove] for sublist in data]

print("Original:", data)
print("After removing", elements_to_remove, ":", result)

Original: [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
After removing [2, 5, 8] : [[1, 3], [4, 6], [7, 9]]

Comparison

Conclusion

Use nested loops when you need to modify the original list in-place and only remove the first occurrence. Use list comprehension for a more concise solution that removes all occurrences and creates a new list structure.