Given a Matrix with variable lengths rows, extract last column.
Input : test_list = [[3, 4, 5], [7], [8, 4, 6], [10, 3]]
Output : [5, 7, 6, 3]
Explanation : Last elements of rows filtered.Input : test_list = [[3, 4, 5], [7], [8, 4, 6]]
Output : [5, 7, 6]
Explanation : Last elements of rows filtered.
Method #1 : Using loop
This is brute way to solve this, we access last element using “-1”, iterate for each row.
Python3
# Python3 code to demonstrate working of # Rear column in Multisized Matrix# Using loop # initializing listtest_list = [[3, 4, 5], [7], [8, 4, 6, 1], [10, 3]] # printing original listprint("The original list is : " + str(test_list)) res = []for sub in test_list: # getting rear element using "-1" res.append(sub[-1]) # printing resultsprint("Filtered column : " + str(res)) |
The original list is : [[3, 4, 5], [7], [8, 4, 6, 1], [10, 3]] Filtered column : [5, 7, 1, 3]
Method #2 : Using list comprehension
This is another way to solve this, in this, we perform above task in similar way, just as a shorthand.
Python3
# Python3 code to demonstrate working of # Rear column in Multisized Matrix# Using list comprehension # initializing listtest_list = [[3, 4, 5], [7], [8, 4, 6, 1], [10, 3]] # printing original listprint("The original list is : " + str(test_list)) # one-liner to solve this problemres = [sub[-1] for sub in test_list] # printing resultsprint("Filtered column : " + str(res)) |
The original list is : [[3, 4, 5], [7], [8, 4, 6, 1], [10, 3]] Filtered column : [5, 7, 1, 3]
Attention geek! Strengthen your foundations with the Python Programming Foundation Course and learn the basics.
To begin with, your interview preparations Enhance your Data Structures concepts with the Python DS Course. And to begin with your Machine Learning Journey, join the Machine Learning – Basic Level Course
