[python] 파이썬 : 조건에 따라 목록을 나누시겠습니까?

내가 가져가 partition출력 하위 시퀀스에서 상대 순서를 유지 하는 게으른 단일 패스 함수를 제안합니다 .

1. 요구 사항

요구 사항은 다음과 같다고 가정합니다.

• 요소의 상대적 순서 유지 (따라서 세트와 사전 없음)
• 모든 요소에 대해 조건을 한 번만 평가하십시오 (따라서 ( i) filter또는 groupby)
• 두 시퀀스 중 하나의 지연 소비를 허용합니다 (사전 계산할 여유가 있다면 순진한 구현도 수용 가능할 것입니다)

2. split도서관

내 partition기능 (아래 소개) 및 기타 유사한 기능으로 작은 라이브러리로 만들었습니다.

• 파이썬 스플릿

PyPI를 통해 정상적으로 설치 가능합니다 :

pip install --user split

조건에 따라 목록 기반을 분할하려면 partitionfunction을 사용하십시오 .

>>> from split import partition
>>> files = [ ('file1.jpg', 33L, '.jpg'), ('file2.avi', 999L, '.avi') ]
>>> image_types = ('.jpg','.jpeg','.gif','.bmp','.png')
>>> images, other = partition(lambda f: f[-1] in image_types, files)
>>> list(images)
[('file1.jpg', 33L, '.jpg')]
>>> list(other)
[('file2.avi', 999L, '.avi')]

3. partition기능 설명

내부적으로 우리는 한 번에 두 개의 서브 시퀀스를 구축해야하므로, 하나의 출력 시퀀스 만 소비하면 다른 하나의 시퀀스도 계산됩니다. 또한 사용자 요청 (상점 처리되었지만 아직 요청되지 않은 요소)간에 상태를 유지해야합니다. 상태를 유지하기 위해 두 개의 이중 끝 큐 ( deques)를 사용합니다.

from collections import deque

SplitSeq 수업은 하우스 키핑을 담당합니다.

class SplitSeq:
    def __init__(self, condition, sequence):
        self.cond = condition
        self.goods = deque([])
        self.bads = deque([])
        self.seq = iter(sequence)

마술은 그 .getNext()방법으로 일어난다 . 그것은 거의 .next()
반복자와 비슷하지만 이번에는 원하는 요소를 지정할 수 있습니다. 장면 뒤에서 거부 된 요소를 버리지 않고 대신 두 개의 대기열 중 하나에 넣습니다.

    def getNext(self, getGood=True):
        if getGood:
            these, those, cond = self.goods, self.bads, self.cond
        else:
            these, those, cond = self.bads, self.goods, lambda x: not self.cond(x)
        if these:
            return these.popleft()
        else:
            while 1: # exit on StopIteration
                n = self.seq.next()
                if cond(n):
                    return n
                else:
                    those.append(n)

최종 사용자는 partition기능 을 사용해야 합니다. 조건 함수와 시퀀스 ( map또는 처럼 filter)를 취하고 두 개의 생성기를 반환합니다. 첫 번째 생성기는 조건이 유지되는 요소의 하위 시퀀스를 작성하고 두 번째 생성기는 상보 하위 시퀀스를 작성합니다. 반복자와 생성기는 길거나 무한한 시퀀스의 지연 분할을 허용합니다.

def partition(condition, sequence):
    cond = condition if condition else bool  # evaluate as bool if condition == None
    ss = SplitSeq(cond, sequence)
    def goods():
        while 1:
            yield ss.getNext(getGood=True)
    def bads():
        while 1:
            yield ss.getNext(getGood=False)
    return goods(), bads()

미래에 부분 적용을 용이하게하기 위해 테스트 함수를 첫 번째 인수로 선택했습니다 (테스트 함수를 첫 번째 인수로 사용하는 방법 map과 유사 함 filter
).

답변

def categorize(func, seq):
    """Return mapping from categories to lists
    of categorized items.
    """
    d = defaultdict(list)
    for item in seq:
        d[func(item)].append(item)
    return d

mylist = [1,2,3,4,5,6,7]
goodvals = [1,3,7,8,9]

myset = set(mylist)
goodset = set(goodvals)

print list(myset.intersection(goodset))  # [1, 3, 7]
print list(myset.difference(goodset))    # [2, 4, 5, 6]

IMAGE_TYPES = set(['.jpg','.jpeg','.gif','.bmp','.png'])