Python – Filter rows with Elements as Multiple of K
Given a Matrix, extract rows with elements multiple of K.
Input : test_list = [[5, 10, 15], [4, 8, 12], [100, 15], [5, 10, 23]], K = 4
Output : [[4, 8, 12]]
Explanation : All are multiples of 4.Input : test_list = [[5, 10, 15], [4, 8, 11], [100, 15], [5, 10, 23]], K = 4
Output : []
Explanation : No rows with all multiples of 4.
Method #1 : Using list comprehension + all()
In this, we check for all elements to be multiple using all(), and iteration of rows occur using list comprehension.
Python3
# Python3 code to demonstrate working of# Access element at Kth index in String# Using list comprehension + all() # initializing string listtest_list = [[5, 10, 15], [4, 8, 3], [100, 15], [5, 10, 23]] # printing original listprint("The original list is : " + str(test_list)) # initializing KK = 5 res = [sub for sub in test_list if all(ele % K == 0 for ele in sub)] # printing resultprint("Rows with K multiples : " + str(res)) |
The original list is : [[5, 10, 15], [4, 8, 3], [100, 15], [5, 10, 23]]
Rows with K multiples : [[5, 10, 15], [100, 15]]
Method #2 : Using filter() + lambda + all()
In this, we perform task of filtering using filter() and lambda function and task of checking for all elements in rows using all().
Python3
# Python3 code to demonstrate working of # Access element at Kth index in String# Using filter() + lambda + all() # initializing string listtest_list = [[5, 10, 15], [4, 8, 3], [100, 15], [5, 10, 23]] # printing original listprint("The original list is : " + str(test_list)) # initializing K K = 5 # using all() to check for all elements being multiples of Kres = list(filter(lambda sub : all(ele % K == 0 for ele in sub), test_list)) # printing result print("Rows with K multiples : " + str(res)) |
The original list is : [[5, 10, 15], [4, 8, 3], [100, 15], [5, 10, 23]]
Rows with K multiples : [[5, 10, 15], [100, 15]]
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
