Python|在矩阵的第 N 列中搜索

有时，在使用Python Matrix 时，我们可能会遇到需要检查元素是否存在的问题。这个问题更简单，但它的一个变体可以是在 Matrix 的特定列中执行检查。让我们讨论一下可以执行此任务的速记。

方法：使用any() + 列表推导
可以使用上述函数的组合来执行此任务，其中我们暗示迭代和支持任务的列表理解，并且any()可用于检查所需字符的任何出现。

# 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