다른리스트의 값을 찾고 그 값을 사전에 추가합니다.

Question

제가 사용하는 지구 화학적 모델의 결과는 단계 당 300ish 화학 종으로 3000+ 단계를 생성합니다.

나는 내가 관심이있는 종과 값의 목록을 가지고있다.

을 어떻게 I에서 다음 내가 관심 (종)이야 종의 목록을 사용하여 데이터 (주)을 반복 관련 종 목록의 두 번째 값을 각 종에 대해 if x=='y':list['x'].append(value) 문을 100 개 입력하지 않고 사전의 키에 추가하십시오.

이 내 코드의 단순화 된 버전입니다 :

for value in main: 
    if value==i for i in species: 
     maindict[i].append(value[2]) 

하지만 분명이 정말 작동하지 않습니다

main =[['a',1,2,3],['b',4,5,6],['c',7,8,9],['a',10,11,12],['b',13,14,15],['c',16,17,18] 
species = ['a', 'b', 'c'] 
maindict={'a':[],'b':[],'c':[]} 

for value in main2: 
    for x in value: 
     if x=='a':maindict['a'].append(value[2]) 
     elif x=='b':maindict['b'].append(value[2]) 
     elif x=='c':maindict['c'].append(value[2]) 

내가 무엇을 찾고 있어요 것은 조금 같은 간단한 일입니다.

출력 :

maindict={'a':[3,12],'b':[6,15],'c':[9,18] 

Answer 1

당신은 메인의 각 항목을 통해 루프가 첫 번째 항목은 maindict에있는 경우, 다음 maindict에 올바른 항목을 추가 확인하실 수 있습니다

main =[['a',1,2,3],['b',4,5,6],['c',7,8,9],['a',10,11,12],['b',13,14,15],['c',16,17,18]] 
species = ['a', 'b', 'c'] 
maindict={'a':[],'b':[],'c':[]} 


for entry in main: 
    if entry[0] in maindict: 
     maindict[entry[0]].append(entry[3]) 
    else: 
     maindict[entry[0]] = [entry[3]] 

print maindict 

>>> {'a': [3, 12], 'c': [9, 18], 'b': [6, 15]} 

Answer 2

점 dict가있는 것은 순차적 인 조회를 피하는 것입니다.

from collections import defaultdict 

def collect(source, species): 
    result = defaultdict(list) 
    for key in species: 
     found = source.get(key) 
     if found: 
      result[key].append(found[2]) 
    return result 


source ={'a':[1,2,3],'b':[4,5,6],'c':[7,8,9], 'd':[0, 0, 0]} 
species = ['a', 'b', 'c', 'e'] 

result = collect(source, species) 
print result 

편집 : 확인 그래서 당신의 소스가 실제로 딕셔너리 아니다 ... 여기리스트 목록에서 작업하는 업데이트 된 버전입니다 (소스는 아마 튜플의 목록이어야합니다 있습니다하지만 다른 점이다) :

from collections import defaultdict 

def collect(source, species): 
    result = defaultdict(list) 
    for row in source: 
     key = row[0] 
     if key in species: 
      result[key].append(row[2]) 
    return result 


source =[['a', 1,2,3], ['b', 4,5,6],['c', 7,8,9], ['d', 0, 0, 0]} 
# a set provides a huge perf improvement over a plain list here 
species = set(['a', 'b', 'c', 'e']) 

result = collect(source, species) 
print result