Знайдіть перший елемент у послідовності, яка відповідає присудку

Я хочу ідіоматичний спосіб знайти перший елемент у списку, який відповідає присудку.

Поточний код досить некрасивий:

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

Я думав про те, щоб змінити його на:

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

Але повинно бути щось більш елегантне ... І було б добре, якби він повертав Noneзначення, а не збільшував виняток, якщо не знайдено відповідності.

Я знаю, що міг би просто визначити функцію на кшталт:

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

Але зовсім несмачно починати наповнювати код такими функціями утиліти (і люди, мабуть, не помітять, що вони вже є, тому вони, як правило, повторюються з часом), якщо є вбудовані модулі, які вже забезпечують те саме.

Щоб знайти перший елемент у послідовності, seqяка відповідає predicate:

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

Або _{( itertools.ifilterна Python 2)} :

next(filter(predicate, seq))

Він піднімається, StopIterationякщо його немає.

Щоб повернутися, Noneякщо такого елемента немає:

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

Або:

next(filter(predicate, seq), None)

Ви можете використовувати вираз генератора зі значенням за замовчуванням, а потім nextйого:

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

Хоча для цього одного вкладиша вам потрібно використовувати Python> = 2.6.

Ця досить популярна стаття далі обговорює це питання: Найясніша функція пошуку в списку Python? .

Я не думаю, що немає нічого поганого в жодному із запропонованих вами запитань.

У власному коді я би реалізував це так:

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

Синтаксис з ()створює генератор, який є більш ефективним, ніж генерувати весь список відразу [].

Відповідь Дж. Ф. Себастьяна є найелегантнішою, але вимагає пітона 2.6, як зазначав фортран.

Для версії Python <2.6, ось найкраще, що я можу придумати:

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

Крім того, якщо вам знадобився список пізніше (список обробляє StopIteration), або вам потрібно більше, ніж просто перший, але все ще не все, ви можете зробити це за допомогою islice:

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

ОНОВЛЕННЯ: Хоча я особисто використовую заздалегідь задану функцію, яку називають first (), яка ловить StopIteration і повертає None, Ось можливе вдосконалення порівняно з наведеним вище прикладом: уникайте використання filter / ifilter:

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