从Python中的双端队列左边弹出一个最多N个条目

Question

我想弹出collections.deque左边最多的limit项。

def popAll(q, limit=None): 
    if limit is None: 
     result = list(q) 
     q.clear() 
    else: 
     result = [] 
     try: 
      for _ in xrange(limit): 
       result.append(q.popleft()) 
     except IndexError: 
      pass 
    return result 

这似乎工作，但似乎......笨重。有没有更“pythonic”的方式？ deque对象不支持切片。

注：我的目的，因为我将要包装一个threading.RLock()访问内部访问deque的，我不需要一个固有的线程安全的答案。

Answer 1

这里是我怎么可能做到这一点：

def popAll(q, limit=None): 
    return [q.popleft() for _ in xrange(min(limit or sys.maxsize, len(q)))] 

或者，也许是这样的：

def popAll(q, limit=None): 
    if limit is None: 
     limit = sys.maxsize 
    return [q.popleft() for _ in xrange(min(limit, len(q)))] 

注意这两个例子中的popAll(q, 0)中的行为。在第一个例子中，它返回整个队列。第二，它返回一个空的列表。

Answer 2

如果它是一个迭代，您可以使用itertools.islice然后取出：

from itertools import islice 

def popAll(q, limit=None): 
    result = list(islice(q, limit)) 
    if limit is None or limit >= len(q): 
     q.clear() 
    else: 
     for _ in range(limit): 
      q.popleft() 
    return result 

万一limit因此不None，我们首先计算我们可以删除（这是最低的项目数q中的项目数量和限制）。在limit为None或大于或等于零的情况下，简单地clear() deque更有效，否则我们遍历元素并一次删除它们。