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


214

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

{
  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,]

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

답변:


278

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'])]

1
collections예를 들어 모듈 아래의 다른 사전 도 이런 방식으로 작동합니다 collections.OrderedDict.
txsaw1

2
오. 대단하다. 그리고 '= []'로 초기화 할 필요가 없습니다. 좋은 물건!
Wilmer E. Henao

1
NameError: name 'a' is not defined
S Andrew

51

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

>>> 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])]

32

당신은 사용할 수 있습니다 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 이전)에서 여전히 유용합니다.


2
이것은 작동하지만 일반적으로 사용 가능한 경우 defaultdict를 사용하는 것이 좋습니다.
David Z

@David, 그렇습니다. setdefault는 가장 화려한 디자인이 아니 었습니다. 죄송합니다. 최고의 선택은 아닙니다. 나는 우리 (파이썬 커미터)가 collections.defaultdict ;;)로 우리의 단체 평판을 상환했다고 생각합니다.
Alex Martelli 2016

@DavidZ, setdefault는 더 유연하기 때문에 defaultdict와 다릅니다. 그렇지 않으면 다른 사전 키에 다른 기본값을 어떻게 지정합니까?
Alex Gidan

@AlexGidan 사실이지만이 질문과 관련이 없습니다.
David Z

이 답변은 OrderedDict와 기본값이 필요한 경우에도 유용합니다.
nimcap

2

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

if d.has_key(scope_item):

와:

if scope_item in d:

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


2

개인적으로 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'}

1

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

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

print(d)
{'1': [1, ], '2':[2, ]}
당사 사이트를 사용함과 동시에 당사의 쿠키 정책개인정보 보호정책을 읽고 이해하였음을 인정하는 것으로 간주합니다.
Licensed under cc by-sa 3.0 with attribution required.