python之二维数组中的查找

最新推荐文章于 2023-04-08 15:46:15 发布

原创最新推荐文章于 2023-04-08 15:46:15 发布 · 6.4k 阅读

0 ·

CC 4.0 BY-SA版权

算法专栏收录该内容

12 篇文章

订阅专栏

本文介绍两种在二维数组中查找特定数值的方法。第一种方法通过将二维数组扁平化为一维数组并使用列表推导式查找目标数值。第二种方法采用双重循环遍历二维数组的每个元素，直接比较目标数值，提高查找效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

# 实现方式一
def find_num(matrix, num):
    """二维数组中的查找"""
    tmp = [i for j in matrix for i in j if i == num]
    return tmp is not None


# 实现方式二
def find_num2(matrix, num):
    """二维数组中的查找"""
    for i in matrix:
        for j in i:
            if j == num:
                return True
    return False

if __name__ == '__main__':

    list2 = [[12, 3, 4], [4, 6, 7, 8]]
    print(find_num2(list2, 6))