Encontre o primeiro elemento em uma sequência que corresponde a um predicado

Eu quero uma maneira idiomática de encontrar o primeiro elemento em uma lista que corresponde a um predicado.

O código atual é bastante feio:

[x for x in seq if predicate(x)][0]

Eu pensei em mudar para:

from itertools import dropwhile
dropwhile(lambda x: not predicate(x), seq).next()

Mas deve haver algo mais elegante ... E seria bom se ele retornasse um Nonevalor em vez de gerar uma exceção se nenhuma correspondência fosse encontrada.

Eu sei que eu poderia apenas definir uma função como:

def get_first(predicate, seq):
    for i in seq:
        if predicate(i): return i
    return None

Mas é bastante insípido começar a preencher o código com funções utilitárias como esta (e as pessoas provavelmente não perceberão que elas já estão lá, então elas tendem a ser repetidas ao longo do tempo) se houver itens incorporados que já forneçam o mesmo.

Para encontrar o primeiro elemento em uma sequência seqque corresponde a predicate:

next(x for x in seq if predicate(x))

Ou _{( itertools.ifilterno Python 2)} :

next(filter(predicate, seq))

Aumenta StopIterationse não houver.

Para retornar Nonese não houver esse elemento:

next((x for x in seq if predicate(x)), None)

Ou:

next(filter(predicate, seq), None)

Você pode usar uma expressão de gerador com um valor padrão e, em seguida, nextela:

next((x for x in seq if predicate(x)), None)

Embora para essa linha única você precise usar Python> = 2.6.

Este artigo bastante popular discute ainda mais esse problema: Função de busca na lista do Python mais limpa? .

Não acho que haja algo errado nas soluções que você propôs na sua pergunta.

No meu próprio código, eu o implementaria assim:

(x for x in seq if predicate(x)).next()

A sintaxe com ()cria um gerador, que é mais eficiente do que gerar toda a lista de uma só vez [].

A resposta de JF Sebastian é mais elegante, mas requer o python 2.6, como fortran apontou.

Para a versão Python <2.6, aqui está o melhor que posso apresentar:

from itertools import repeat,ifilter,chain
chain(ifilter(predicate,seq),repeat(None)).next()

Como alternativa, se você precisar de uma lista posteriormente (a lista lida com o StopIteration) ou se precisar de mais do que apenas o primeiro, mas ainda não todos, poderá fazê-lo com o islice:

from itertools import islice,ifilter
list(islice(ifilter(predicate,seq),1))

ATUALIZAÇÃO: Embora eu pessoalmente esteja usando uma função predefinida chamada first () que captura uma StopIteration e retorna None, aqui está uma possível melhoria em relação ao exemplo acima: evite usar filter / ifilter:

from itertools import islice,chain
chain((x for x in seq if predicate(x)),repeat(None)).next()