목록의 다음 요소에 따라 첫 번째 요소의 범위를 가져 오는 중

Question

다음과 관련하여 어려움을 겪고 있습니다. 나는 간단한 비교와 함께 할 시도 요소 (2)와 (3)

uniqdo = [ (['A1'],['A2']), (['B1'],['B2']),([],['A2']) ] 
dorange = [ "1280-1276,1272","1274","1270" ] 

으로 분류 소트

dolist = [(1280, ['A1'], ['A2']), (1278, ['A1'], ['A2']), (1276, ['A1'], ['A2']), (1274, ['B1'], ['B2']), (1272, ['A1'], ['A2']), (1270, [], ['A2'])] 

가 지금은 목록을 갖고 싶어하지만 코드는 여러 가지로 매우 긴됩니다 : 기본적으로 나는 목록을 가지고 테스트하고 좀 지저분 해 보입니다. 이 작업을 합리적인 속도로 수행 할 수있는 라이브러리 함수가 있어야합니다.

Answer 1

그것은 itertools.groupby는 도움이되는 다음과 같습니다

>>> dolist = [ (1280,['A1'],['A2']),(1278,['A1'],['A2']),(1276,['A1'],['A2']),(1274,['B1'],['B2']),(1272,['A1'],['A2']) ] 
>>> from itertools import groupby 
>>> [[v, [i for i,*_ in g]] for v, g in groupby(dolist, key= lambda l: (l[1][0], l[2][0]))] 
[[('A1', 'A2'), [1280, 1278, 1276]], [('B1', 'B2'), [1274]], [('A1', 'A2'), [1272]]] 

당신이 원하는 일에 위의 데이터 구조를 변환하는 데 어렵지 않을 것이다.

여기가 시작되었습니다. 파이썬 목록을 dict 키로 사용할 수 없기 때문에 목록을 입력으로 남겨 둘 수 없습니다. 그래서 get_value 반환하는 대신 빈리스트의 None : 1278`가 흡수`경우

from itertools import groupby 

dolist = [(1280, ['A1'], ['A2']), (1278, ['A1'], ['A2']), (1276, ['A1'], ['A2']), (1274, ['B1'], ['B2']), (1272, ['A1'], ['A2']), (1270, [], ['A2'])] 
ranges = {} 


def get_value(l): 
    if l: 
     return l[0] 
    else: 
     return None 


def get_values(t): 
    return (get_value(t[1]), get_value(t[2])) 

for v, g in groupby(dolist, get_values): 
    ids = [str(t[0]) for t in g] 
    if len(ids) > 1: 
     range_str = ids[0] + '-' + ids[-1] 
    else: 
     range_str = ids[0] 
    ranges.setdefault(v, []).append(range_str) 

print(ranges) 
# {('A1', 'A2'): ['1280-1276', '1272'], ('B1', 'B2'): ['1274'], (None, 'A2'): ['1270']}