Skip to content
Related Articles

Related Articles

Improve Article

Python | Search in Nth Column of Matrix

  • Last Updated : 27 Sep, 2019

Sometimes, while working with Python Matrix, we can have a problem in which we need to check if an element is present or not. This problem is simpler, but a variation to it can be to perform a check in particular column of Matrix. Let’s discuss a shorthand by which this task can be performed.

Method : Using any() + list comprehension
This task can be performed using combination of above functions in which we imply list comprehension for iteration and support tasks and any() can be used to check for any occurrence of the desired character.

 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




# Python3 code to demonstrate working of
# Search in Nth Column of Matrix
# Using any() + list comprehension
  
# initializing list 
test_list = [[1, 4, 5], [6, 7, 8], [8, 3, 0]]
  
# printing list 
print("The original list : " + str(test_list))
  
# initializing N 
N = 1 
  
# initializing num
ele = 3
  
# Search in Nth Column of Matrix
# Using any() + list comprehension
res = any(sub[N] == ele for sub in test_list)
  
# Printing result
print("Does element exists in particular column : " + str(res))
Output :
 
The original list : [[1, 4, 5], [6, 7, 8], [8, 3, 0]]
Does element exists in particular column : True
My Personal Notes arrow_drop_up
Recommended Articles
Page :