Python | Buscar en la enésima columna de la array

A veces, mientras trabajamos con Python Matrix, podemos tener un problema en el que necesitamos verificar si un elemento está presente o no. Este problema es más simple, pero una variación puede ser realizar una verificación en una columna particular de Matrix. Analicemos una abreviatura mediante la cual se puede realizar esta tarea.

Método: Usar any()+ comprensión de lista
Esta tarea se puede realizar usando una combinación de las funciones anteriores en las que implicamos comprensión de lista para iteración y tareas de soporte y any()se puede usar para verificar si aparece el carácter deseado.

# 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))

 
The original list : [[1, 4, 5], [6, 7, 8], [8, 3, 0]]
Does element exists in particular column : True