삶은 여행

python dictionary value list append

​

딕셔너리는 키/값 (key / value)으로 이루어져있다. 파이썬3.7+부터는 입력한 대로 순서가 유지된다.

내부적으로는 해시테이블로 구현되어있다.

​

key는 변화하면 안되기 때문에 가변적인 list가 key값으로는 들어갈 수 없지만 tuple은 key값으로 사용할 수 있다.

value는 tuple, list 관계없이 전부 가능하다.

​

딕셔너리 주요 연산 시간 복잡도

대부분의 연산이 O(1)로 처리가 가능해 우수한 자료형으로 꼽힌다.

​

리스트는 내부 요소를 조회할 때 인덱스의 순서로 (list[0]과 같이) 조회하지만

딕셔너리는 key 값을 직접 입력해야 value 값을 조회할 수 있다.

​

딕셔너리에는 defaultdict 라는 모듈이 있는데 딕셔너리에 값을 삽입하는 데에 있어

테크니컬적인 부분에서 유연하게 딕셔너리를 다루기에 유용하다.

딕셔너리의 defaultdict 객체

리스트에서 존재하지 않는 인덱스를 조회할 경우 IndexError가 발생하듯이

딕셔너리에서 존재하지 않는 키를 조회할 경우 KeyError가 발생하는데

KeyError의 에러 메시지를 출력하는 대신 디폴트 값을 기준으로 해당 키에 대한 딕셔너리 아이템을 생성해준다.

>>> d = defaultdict(int)
>>> d['A'] = 0
>>> d['B'] = 1
>>> d['C'] += 1

을 실행할 경우, 기존 딕셔너리 d 는 {'A' : 0 , 'B' : 1}의 형태로 키값 'C'가 존재하지 않아 4행에서 keyerror가 나와야하지만

defaultdict 객체를 이용했기 때문에 디폴트값 0을 기준으로 하여 0 + 1의 연산이 이루어져 d는 다음과 같아진다.

{'A' : 0 , 'B' : 1, 'C' : 1}

defaultdict(int)뿐만 아니라 defaultdict(list)를 이용하면 value값에 리스트를 추가할 수도 있다.

해시테이블 구조로 정리해야하는 입력값이 이중리스트로 주어졌을 때 딕셔너리에 값을 삽입할 수 있는 방법은 다음과 같다.

예시

입력치 : 리스트 [ [ '카테고리A' , 'A1'] , ['카테고리B' , 'B1'] , ['카테고리A' , 'A2'] ]

기대하는 결과값 : 딕셔너리 { '카테고리A' : ['A1' , 'A2'] , '카테고리B' : ['B1'] }

>>> for i in range(len(리스트)):
>>>     딕셔너리[리스트[i][1]] += [리스트[i][0]]

위와 같이 카테고리별로 정리한 딕셔너리에서

각 카테고리 별 요소가 몇 개나 있는지 key별 요소 개수를 확인하고 싶을 때는 다음과 같다.

>>> lengths = [len(v) for v in dictionary.values()]
>>> lengths
[2, 1]

문제 출처 https://programmers.co.kr/learn/courses/30/lessons/42578

문제 설명

스파이들은 매일 다른 옷을 조합하여 입어 자신을 위장합니다.

스파이가 가진 의상들이 담긴 2차원 배열 clothes가 주어질 때 서로 다른 옷의 조합의 수를 return 하도록 solution 함수를 작성해주세요.

​

제한사항

clothes의 각 행은 [의상의 이름, 의상의 종류]로 이루어져 있습니다.

스파이가 가진 의상의 수는 1개 이상 30개 이하입니다.

같은 이름을 가진 의상은 존재하지 않습니다.

clothes의 모든 원소는 문자열로 이루어져 있습니다.

모든 문자열의 길이는 1 이상 20 이하인 자연수이고 알파벳 소문자 또는 '_' 로만 이루어져 있습니다.

스파이는 하루에 최소 한 개의 의상은 입습니다.

휴 40분 걸렸다

수학문제에 더 가까운 것 같은 문제.

​

의상카테고리가 하나일 때는 그 카테고리에 있는 개수를 출력하기만 하면 되지만

카테고리의 개수가 3개 이상으로 점점 늘어날수록 한 카테고리의 의상은 아예 선택하지 않는 경우까지 고려해야하다보니 구현을 어떻게 해야할지 통 감이 안와서 시간이 많이 걸렸다.

​

딕셔너리랑 각 카테고리별 요소 개수를 저장하는 리스트구현까지는 성공했는데

그 이후에 답을 얻어내는 과정을 모르겠어서 결국 답을 치팅했다.

(요소개수 + 1) 의 모든 곱이라니..

치팅하고 나서 공식으로 풀어보니까 위와 같이 기가막히게 똑같이 나왔다.

그래도 딕셔너리 밸류에 리스트를 넣는 등 테크니컬적인 부분에서는 공부가 됐다.

def solution(clothes):
    from collections import defaultdict
    d = defaultdict(list)
    def multi(arr):
        ans = 1
        for i in arr:
            ans *= (i+1)
        return ans
    for i in range(len(clothes)):
        d[clothes[i][1]] += [clothes[i][0]]
    lengths = [len(v) for v in d.values()]
    if len(lengths)==1:
        return lengths[0]
    else: return (multi(lengths) -1)