một cách khác nữa để thể hiện vòng lặp bị ràng buộc có điều kiện?

Question

Tôi có một vòng lặp muốn thực thi để cạn kiệt hoặc cho đến khi đạt đến một số giới hạn do người dùng chỉ định. Tôi đã có một cấu trúc trông xấu nhưng tôi dường như không thể tìm thấy một cách thanh lịch hơn để thể hiện nó; có ai không?

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 

Làm tổ thánh! Có phải là một cách tốt hơn. Với mục đích của một ví dụ làm việc, xrange được sử dụng trong đó tôi thường có một trình vòng lặp có độ dài hữu hạn nhưng không xác định (và vị ngữ đôi khi trả về Sai).

Answer 1

Có lẽ một cái gì đó như thế này sẽ tốt hơn một chút:

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

Tôi sẽ xem xét tốt thư viện itertools. Sử dụng điều đó, tôi nghĩ rằng bạn muốn có một cái gì đó giống như ...

# 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

Tôi muốn bắt đầu với

if limit is None: return 

vì không có gì có thể xảy ra với limit khi nó bắt đầu là None (nếu không có tác dụng phụ mong muốn trong lặp lại và trong tính toán của predicate - nếu có, sau đó, trong trường hợp này bạn chỉ có thể làm for i in xrange(10**5): predicate(i)).

Nếu limit không phải là None, sau đó bạn chỉ muốn thực hiện max(limit, 1) tính toán của predicate đó là đúng sự thật, vì vậy một itertools.islice của một itertools.ifilter sẽ làm gì:

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 4

gì bạn muốn làm dường như hoàn toàn phù hợp cho một while loop:

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 

Vòng lặp dừng nếu max hoặc limit đạt 0.

Answer 5

Bạn nên loại bỏ các ifs lồng nhau:

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

Answer 6

Um. Theo tôi hiểu, predicate chỉ tính toán trong phân khúc và bạn hoàn toàn bỏ qua giá trị trả lại của nó, phải không?

Đây là một mất:

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 

Ngoài ra, loại bỏ các không cần thiết return True từ cuối predicate.