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

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
    

ᕕ( ᐛ )ᕗ
로그인이 필요합니다

작성한 답변에 다른 개발자들이 댓글을 작성하거나 댓글에 좋아요/싫어요를 할 수 있기 때문에 계정을 필요로 합니다.