考虑这个简单的例子
mylist = [-1,-2,3,4,5,6]
for idx, el in enumerate(mylist):
if el > 0:
myidx, myel = idx, el
break
myidx, myel
Out[20]: (2, 3)
我感兴趣的是在python列表中查找与特定条件匹配的第一个索引和相应的第一个元素(这里,这只是>;0)
在上面的代码中,我使用enumerate
循环元素,然后使用if子句查找正确的元素。我觉得这很麻烦。有更好的方法吗?例如,使用本机python函数
谢谢
Tags:
像这样的方法应该会奏效:
为了找到所有模式,我们可以使用python内置函数
你可以在列表中完成。这基本上与您的代码相同,但压缩为一行,它构建了一个符合条件的结果列表
第一种方法获得所有匹配项
另一种方法是使用可能更快的生成器表达式,然后将其解包。这是第一个
这将循环列表并检查条件,然后将索引和元素放入元组中。在括号内这样做会使它变成一个生成器,速度快得多,因为它实际上不需要在内存中保存所有内容,它只会根据需要生成响应。使用
next()
可以遍历它们。因为我们在这里只使用next()一次,所以它只生成第一个匹配项。然后我们用*
解压它由于这里还有另外两个有效答案,我决定使用timeit模块对每个答案进行计时并发布结果。为了清楚起见,我还对OP的方法进行了计时。以下是我的发现:
您可以像这样使用
lambda
和filter
的组合:说明:
filter()
函数提供了一种优雅的方式来过滤掉所有元素,它将函数和列表作为参数。这里是lambda
和mylist
。因为您想要获得相应的索引,所以我们需要使用enumerate
来结束enumerate(mylist)
基本上,
enumerate(mylist)
返回索引和相应值的元组。这里的条件是值和0
之间的比较,这就是为什么我们得到el[1]
而不是el[0]
来与0
进行比较结果将被转换为
list
。此列表包括满足我们条件的所有对(index, value)
。这里我们想要得到第一对,这就是为什么我们在末尾有[0]
输出:
相关问题 更多 >
编程相关推荐