[Snippet] 자주 사용하는 List, Dictionary 기능

코드 작성을 하다 보면 항상 잊어버리거나 기억이 잘 나지 않는 기능들이 있습니다. 우리나라 말을 할 때도 가끔 단어가 생각이 안 날 때처럼 말이죠 ㅎㅎ 괜히 코딩하는 사람들이 구글을 "신"이라고 하는 게 아닙니다. ChatGPT가 마이크로소프트의 검색엔진 Bing에 잘 탑재만 된다면 앞으로 개발 생산성은 더더욱 향상될 것 같습니다. 

 

얘기가 조금 샜는데, 회사에서나 취미로나 코딩을 할 때 자주 까먹는 기능들이 있습니다. List안에 있는 List를 하나의 List로 병합한다던지, Dictionary 안에 있는 키 값들을 어떻게 정렬하는지 등 자주 사용하지만 매번 외우는 데는 실패하는 코드들이 있습니다. 따라서 이번 포스트에서는 자주 사용하는 List와 Dictionary 기능들에 대해 정리하려고 합니다.

코드 스니펫!

1. 리스트 안에 리스트들을 하나의 리스트로 합치기

크게 네 가지 방법이 있습니다. 

`For`문을 한 번 혹은 두 번 사용하여 list의 메서드인. append() 와. extend()를 사용해서 해결할 수 있고, List comprehension을 사용해서 한 줄로 정리할 수도 있습니다. 또한 내장함수인 itertools의 chain을 이용해서 해결할 수도 있습니다. 어느 것이 더 낫다고 말하기에는 어렵습니다. 어떤 방법은 코드가 무엇을 하는지 더 쉽게 알아볼 수 있다는 장점이 있고, 어떤 방법은 코드가 차지하는 공간이 적어져서 관리하고 보기에 더 편합니다. 

# 서브리스트
sublist_1 = ['카지노','오징어게임','더 글로리']
sublist_2 = ['재벌집 막내아들', '카지노', '소년심판']
sublist_3 = ['낮과밤', '웬즈데이', '수리남', '미생']
sublist_4 = ['종이의집','우리들의 블루스', '멜로가체질']

# 여러 개의 서브리스트로 된 리스트
lst = [sublist_1, sublist_2, sublist_3, sublist_4]



# 모든 요소를 하나의 리스트로 합치기
## 방법 1: List Comprehension
def method1(lst):
    return [item for sublist in lst for item in sublist]


## 방법 2: Looping (코드가 길어서 지저분해지지만, readability는 더 좋음)
def method2(lst):
    result = []
    for sublist in lst:
        for item in sublist:
            result.append(item)
    return result


# 방법 3: extend 사용
def method3(lst):
    result = []
    for sublist in lst:
        result.extend(sublist)
    return result


## 방법 4: 내장함수 itertools.chain() 사용 --> sublist가 많아지면 가장 빠른 방법입니다
from itertools import chain

def method4_a(lst):    
    return list(chain(*lst))  

def method4_b(lst):
    return list(chain.from_iterable(lst)) # 혹은 이렇게도 가능

 

2. 문자열로 표현된 리스트, 딕셔너리를 읽는 방법

리스트나 딕셔너리를 pandas.DataFrame에 저장하는 경우가 있는데, 꼭 불러오면 죄다 문자열 형태로 되어 있어 난감할 때가 있습니다. 그럴 때 사용하시면 됩니다.

# Built-in 내장함수 사용
from ast import literal_eval

string = '["이","데이터는","리스트","일까요?","문자열","일까요?"]'
lst = literal_eval(string)

print(lst)

['이', '데이터는', '리스트', '일까요?', '문자열', '일까요?']

 

 

3. 딕셔너리를 Key 값 혹은 Value 값으로 정렬하기

딕셔너리에 저장된 값을 정렬하고 싶을 때가 있습니다. 하지만, 딕셔너리의 경우에는 기본적으로 순서라는 개념이 없기 때문에 가끔 코드를 짜다가 막히는 경우가 있습니다. 아래와 같이 OrderedDict를 사용하면 편합니다.

from collections import OrderedDict
from operator import itemgetter

apt = {'Mapo': 12, 'Gangnam': 22,
        'Songpa': 16, 'Songdo': 8, 'Hannam': 20}

# Key 값으로 정렬 (오름차순)
result1 = OrderedDict(sorted(apt.items()))   

# Value 값으로 정렬 (오름차순)
result2 = OrderedDict(sorted(apt.items(), key=itemgetter(1)))


# Key 값으로 정렬 (내림차순)
# result3 = OrderedDict(sorted(apt.items(), reverse=True))   

# Value 값으로 정렬 (내림차순)
# result4 = OrderedDict(sorted(apt.items(), key=itemgetter(1), reverse=True))

result1 

key 값의 오름차순인 a, b, c, d.. 순서로 정렬된 것을 볼 수 있습니다.

OrderedDict([('Gangnam', 22), ('Hannam', 20), ('Mapo', 12), ('Songdo', 8), ('Songpa', 16)])

result2

value 값의 오름차순인 1, 2, 3, 4.. 순서로 정렬된 것을 볼 수 있습니다.

OrderedDict([('Songdo', 8), ('Mapo', 12), ('Songpa', 16), ('Hannam', 20), ('Gangnam', 22)])

 

막상 쓰려니까 생각이 많이 나지는 않네요...

이 포스트에 조금씩 추가해 나가도록 하겠습니다.