[python] 리스트의 사전을 만드는 파이썬

PYTHON / 글쓴이

값이 목록 인 사전을 만들고 싶습니다. 예를 들면 다음과 같습니다.

{
  1: ['1'],
  2: ['1','2'],
  3: ['2']
}

만약 내가한다면:

d = dict()
a = ['1', '2']
for i in a:
    for j in range(int(i), int(i) + 2):
        d[j].append(i)

d […]가 목록이 아니기 때문에 KeyError가 발생합니다. 이 경우 a를 할당 한 후 다음 코드를 추가하여 사전을 초기화 할 수 있습니다.

for x in range(1, 4):
    d[x] = list()

더 좋은 방법이 있습니까? 두 번째 for루프 가 될 때까지 필요한 키를 모른다고 가정 해 봅시다. 예를 들면 다음과 같습니다.

class relation:
    scope_list = list()
...
d = dict()
for relation in relation_list:
    for scope_item in relation.scope_list:
        d[scope_item].append(relation)

대안은 다음 교체 될 것입니다 

d[scope_item].append(relation)

와 

if d.has_key(scope_item):
    d[scope_item].append(relation)
else:
    d[scope_item] = [relation,]

이것을 처리하는 가장 좋은 방법은 무엇입니까? 이상적으로 추가하면 “그냥 작동”할 것입니다. 처음 목록을 만들 때 모든 키를 모르더라도 빈 목록의 사전을 원한다고 표현하는 방법이 있습니까?

답변

defaultdict 를 사용할 수 있습니다 :

>>> from collections import defaultdict
>>> d = defaultdict(list)
>>> a = ['1', '2']
>>> for i in a:
...   for j in range(int(i), int(i) + 2):
...     d[j].append(i)
...
>>> d
defaultdict(<type 'list'>, {1: ['1'], 2: ['1', '2'], 3: ['2']})
>>> d.items()
[(1, ['1']), (2, ['1', '2']), (3, ['2'])]

답변

다음과 같은 목록 이해로 빌드 할 수 있습니다.

>>> dict((i, range(int(i), int(i) + 2)) for i in ['1', '2'])
{'1': [1, 2], '2': [2, 3]}

그리고 질문의 두 번째 부분에는 defaultdict를 사용하십시오.

>>> from collections import defaultdict
>>> s = [('yellow', 1), ('blue', 2), ('yellow', 3), ('blue', 4), ('red', 1)]
>>> d = defaultdict(list)
>>> for k, v in s:
        d[k].append(v)

>>> d.items()
[('blue', [2, 4]), ('red', [1]), ('yellow', [1, 3])]

답변

당신은 사용할 수 있습니다 setdefault:

d = dict()
a = ['1', '2']
for i in a:
    for j in range(int(i), int(i) + 2):
        d.setdefault(j, []).append(i)

print d  # prints {1: ['1'], 2: ['1', '2'], 3: ['2']}

다소 이상한 이름의 setdefault함수는 “이 키로 값을 가져 오거나 해당 키가 없으면이 값을 추가 한 다음 반환합니다.”라고 말합니다.

다른 사람들이 올바르게 지적했듯이 defaultdict더 좋고 현대적인 선택입니다. setdefault이전 버전의 Python (2.5 이전)에서 여전히 유용합니다.

답변

귀하의 질문에 이미 답변되었지만 IIRC는 다음과 같은 줄을 바꿀 수 있습니다.

if d.has_key(scope_item):

와:

if scope_item in d:

즉, 해당 구성의 d참조 d.keys()입니다. 때로는 defaultdict최상의 옵션이 아닙니다 (예 : else위와 연결된 후 여러 줄의 코드를 실행하려는 경우 if) in. 구문을 읽기가 더 쉽습니다.

답변

개인적으로 JSON을 사용하여 사물을 문자열로 변환하고 다시 변환합니다. 내가 이해하는 줄.

import json
s = [('yellow', 1), ('blue', 2), ('yellow', 3), ('blue', 4), ('red', 1)]
mydict = {}
hash = json.dumps(s)
mydict[hash] = "whatever"
print mydict
#{'[["yellow", 1], ["blue", 2], ["yellow", 3], ["blue", 4], ["red", 1]]': 'whatever'}

답변

쉬운 방법은 다음과 같습니다

a = [1,2]
d = {}
for i in a:
  d[i]=[i, ]

print(d)
{'1': [1, ], '2':[2, ]}

답변