dict에 key가 있는지확인 한 후 값을 증가시키려고 하는데 조언좀 해주세요


발생하는 문제 및 실행환경

dict가 이미 있고, 제가 어떤 key를 갖고있을 때, dict[key]None인지 아닌지 확인하고,

  1. None이 아니면 값을 증가시킴
  2. 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
  • 2016년 02월 15일에 작성됨

조회수 154


1 답변


좋아요
0
싫어요
채택취소하기

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
  • 2016년 02월 15일에 작성됨

로그인이 필요한 기능입니다.

Hashcode는 개발자들을 위한 무료 QnA사이트 입니다. 작성한 답변에 다른 개발자들이 댓글을 작성하거나 좋아요/싫어요를 할 수 있기 때문에 계정을 필요로 합니다.
► 로그인
► 계정만들기
Close