Python - 检查元素是否在列表中(pythonic方式)

时间:2015-01-21 21:46:50

标签: python arrays list search

假设我有一个列表“图”初始化如下。

graph = [[1, 2, 3], [2, 3, 4], [3, 5, 7]]

我如何确定1是否在图表中?

是否有比做

之类的更简单,更优化的方式
in_graph = False
for row in graph:
    if 1 in row:
        in_graph = True
        break

谢谢,

滚刀

2 个答案:

答案 0 :(得分:1)

尝试

any(x in row for row in graph)

其中x是您要查找的元素。

答案 1 :(得分:0)

您可以将any与生成器表达式一起使用。

any(1 in g for g in graph)