dict에 key가 있는지확인 한 후 값을 증가시키려고 하는데 조언좀 해주세요
조회수 7079회
발생하는 문제 및 실행환경
dict가 이미 있고, 제가 어떤 key
를 갖고있을 때, dict[key]
가 None
인지 아닌지 확인하고,
None
이 아니면 값을 증가시킴None
이면 1로 초기화시켜주려고 합니다.
밑에 코드처럼 하면 될 줄 알았는데 안되서 여기에 질문 드립니다.
에러내용
if (my_dict[key] != None):
KeyError: 3
소스코드
my_dict = {}
if (my_dict[key] != None):
my_dict[key] = 1
else:
my_dict[key] += 1
1 답변
-
dict에서 없는 키를 찾는 경우(
my_dict[key]
)는None
이 return되는 게 아니라KeyError
를 raise합니다.그래서 코드를
if key in my_dict: my_dict[key] += 1 else: my_dict[key] = 1
이렇게 바꾸면 원하시는 대로 작동할 겁니다.
그렇지만 이 기능은 이미 파이썬 표준 함수로 있기 때문에 collections.defaultdict 쓰는 걸 추천합니다.(2.5+ 가능)
from collections import defaultdict my_dict = defaultdict(int) my_dict[key] += 1
댓글 입력