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

조회수 7079회

발생하는 문제 및 실행환경

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
    

답변을 하려면 로그인이 필요합니다.

프로그래머스 커뮤니티는 개발자들을 위한 Q&A 서비스입니다. 로그인해야 답변을 작성하실 수 있습니다.

(ಠ_ಠ)
(ಠ‿ಠ)