более питонический способ выразить условно ограниченную петлю?

Question

У меня есть цикл, который хочет выполнить до полного исчерпания или до достижения определенного пользователем предела. У меня есть конструкция, которая выглядит плохо, но я не могу найти более элегантный способ выразить ее; есть один?

def ello_bruce(limit=None):
    for i in xrange(10**5):
        if predicate(i):
            if not limit is None:
                limit -= 1
                if limit <= 0:
                   break

def predicate(i):
    # lengthy computation
    return True

Святое вложение! Там должен быть лучший путь. Для целей рабочего примера используется xrange, где у меня обычно есть итератор конечной, но неизвестной длины (а предикат иногда возвращает False).

Answer 1

Может быть, что-то вроде этого будет немного лучше:

from itertools import ifilter, islice

def ello_bruce(limit=None):
    for i in islice(ifilter(predicate, xrange(10**5)), limit):
        # do whatever you want with i here

Answer 2

Я бы внимательно посмотрел на библиотеку itertools .Используя это, я думаю, у вас будет что-то вроде ...

# From the itertools examples
def tabulate(function, start=0):
    return imap(function, count(start))
def take(n, iterable):
    return list(islice(iterable, n))

# Then something like:
def ello_bruce(limit=None):
  take(filter(tabulate(predicate)), limit)

Answer 3

Вы должны удалить вложенные ifs:

if predicate(i) and not limit is None:
    ...

Answer 4

Я бы начал с

if limit is None: return

, поскольку ничего не может случиться с limit, когда он начинается как None (если в итерации и при вычислении predicate нет желательных побочных эффектов - если таковые имеются, то в этом случае вы можно просто сделать for i in xrange(10**5): predicate(i)).

Если limit не None, то вы просто хотите выполнить max(limit, 1) вычислений predicate, которые являются истинными, поэтому itertools.islice из itertools.ifilter будет делать:

import itertools as it

def ello_bruce(limit=None):
    if limit is None:
        for i in xrange(10**5): predicate(i)
    else:
        for _ in it.islice(
          it.ifilter(predicate, xrange(10**5),
          max(limit, 1)): pass

Answer 5

Um. Насколько я понимаю, predicate просто вычисляет в сегментах, и вы полностью игнорируете его возвращаемое значение, верно?

Это еще один дубль:

import itertools

def ello_bruce(limit=None):
    if limit is None:
        limiter= itertools.repeat(None)
    else:
        limiter= xrange(limit)

    # since predicate is a Python function
    # itertools looping won't be faster, so use plain for.
    # remember to replace the xrange(100000) with your own iterator
    for dummy in itertools.izip(xrange(100000), limiter):
        pass

Также удалите ненужные return True с конца predicate.

Answer 6

То, что вы хотите сделать, кажется идеально подходящим для цикла while:

def ello_bruce(limit=None):
    max = 10**5
    # if you consider 0 to be an invalid value for limit you can also do
    # if limit:
    if limit is None: 
        limit = max

    while max and limit:
        if predicate(i):
            limit -= 1
        max -=1

Цикл останавливается, если max или limit достигает нуля.