Python: Remove a specified column from a given nested list

Python List: Exercise - 107 with Solution

Write a Python program to remove a specified column from a given nested list.

Sample Solution:

Python Code:

def remove_column(nums, n):
   for i in nums: 
    del i[n] 
   return nums

list1 = [[1, 2, 3], [2, 4, 5], [1, 1, 1]]
n = 0
print("Original Nested list:")
print(list1)
print("After removing 1st column:")
print(remove_column(list1, n))

list2 = [[1, 2, 3], [-2, 4, -5], [1, -1, 1]]
n = 2
print("\nOriginal Nested list:")
print(list2)
print("After removing 3rd column:")
print(remove_column(list2, n))

Sample Output:

Original Nested list:
[[1, 2, 3], [2, 4, 5], [1, 1, 1]]
After removing 1st column:
[[2, 3], [4, 5], [1, 1]]

Original Nested list:
[[1, 2, 3], [-2, 4, -5], [1, -1, 1]]
After removing 3rd column:
[[1, 2], [-2, 4], [1, -1]]

Pictorial Presentation:

Flowchart:

Visualize Python code execution:

The following tool visualize what the computer is doing step-by-step as it executes the said program:

Python Code Editor:

Have another way to solve this solution? Contribute your code (and comments) through Disqus.

Previous: Write a Python program to count integer in a given mixed list.
Next: Write a Python program to extract a specified column from a given nested list.