Python - 하나의 요소로 그룹 목록을 나열하십시오.

Question

파일 입력에서 데이터를 처리하려고합니다. 한 줄에는 공백으로 구분 된 3 개의 값이 들어 있습니다. 두 번째 값으로 그룹화 할 목록에 추가하려고합니다. 그럼. I는 입력이 있습니다

qwe rty 12 
asd fgh 34 
zxc rty 96 

와 나는이 같은 변수에 저장할 :

variable = 
[[[qwe, rty, 12], [zxc, rty, 96]], 
[[asd, fgh, 34]]] 

이 내가 이런 식으로 액세스 할 수 있도록이다

variable[0] #this should be [[qwe, rty, 12], [zxc rty, 96]] 
variable[1] #this should be[[asd, fgh, 34]] 

나는 노력하고있다

f = open('input_1.txt') 
values = [] #to keep track of which values have occured before 
data = [] 
for line in f: 
    ldata = lineprocess(line) #this transforms the raw data to [qwe, rty, 12] etc. 
    if ldata[1] in values: 
     data[values.index(ldata[1])].append(ldata) 
    elif ldata[1] not in values : 
     values.append(ldata[1]) 
     data.append(ldata) 

이것은 회전과 같은 목록 :

[['qwe', 'rty', 12, ['zxc', 'rty', 96]], 
['asd', 'fgh', 34]] 

내가 대신

[[['qwe', 'rty', 12], ['zxc', 'rty', 96]], 
[['asd', 'fgh', 34]]] 

를 얻으려면 어떻게해야합니까?

Answer 1

data에는 목록뿐만 아니라 목록이 포함되어야합니다.

f = open('input_1.txt') 
values = [] #to keep track of which values have occured before 
data = [] 
for line in f: 
    ldata = lineprocess(line) #this transforms the raw data to [qwe, rty, 12] etc. 
    if ldata[1] in values: 
     data[values.index(ldata[1])].append(ldata) 
    else: 
     values.append(ldata[1]) 
     data.append([ldata]) 

고려 : 당신이 사전을 원하지 않는 경우

a = [1,2,3] 
b = [4,5,6] 
a.append(b) 
print a # [1, 2, 3, [4, 5, 6]] 
c = [[1,2,3]] 
c.append(b) 
print C# [[1, 2, 3], [4, 5, 6]] 

Answer 2

, 당신은 groupby

from itertools import groupby 
from operator import itemgetter 

with open(...) as lines: 
    parsed_lines = map(lineprocess, lines) # I'm using your `lineprocess` 
    second_item = itemgetter(1) 
    groups = groupby(sorted(parsed_lines, key=second_item), second_item) 
    result = [list(group) for predicate, group in groups] 

사용할 수있는이 당신의 O(n^2)보다 더 O(nlogn) 평균 경우 성능을 가지고 있습니다. 여전히 사전 기반 솔루션은 O(n)입니다.