[Snippet] 상황별 Tuple, List 변환 방법

최근에 NER 태깅 작업을 하면서 tuple과 list를 변환하는 작업을 많이 반복하게 되어 포스팅을 작성하게 되었습니다.

매번 구글링을 통해서 코드를 찾아내서 복붙하고 수정하는 과정 조차 귀찮아지게 되어서 이번 기회에 아예 정리를 해보려고 합니다. 

1. Tuple로 구성된 List를 하나의 List로 변환

ex. [('A',100), ('B',200), ('C',300), ('D',400), ('E',500)]  =>  ['A', 100, 'B', 200, 'C', 300, 'D', 400, 'E', 500]

 

# Tuple로 된 list
lst = [('A',100), ('B',200), ('C',300), ('D',400), ('E',500)]

# 모든 element를 하나의 list로 변환
## 방법 1: List comprehension
new_lst1 = [e for tupl in lst for e in tupl]

## 방법 2: for loop
new_lst2 = []
for x,y in lst:
    new_lst2.extend([x,y])
    
print(new_lst1)
print(new_lst2)

['A', 100, 'B', 200, 'C', 300, 'D', 400, 'E', 500]
['A', 100, 'B', 200, 'C', 300, 'D', 400, 'E', 500]

 

for문을 활용해서 할 수도 있지만 저는 list comprehension 방식을 선호합니다. 코드를 한줄만으로 짤 수 있기 때문입니다.하지만 다른 사람이 이해할 때는 가독성이 상대적으로 떨어지기 때문에 상황에 맞게 사용해야 할 것 같습니다.

 

 

2. List에 있는 element를 순차적으로 2개 씩 tuple로 구성하여 변환

1번의 반대 상황입니다.

ex. ['A', 100, 'B', 200, 'C', 300, 'D', 400, 'E', 500] => [('A',100), ('B',200), ('C',300), ('D',400), ('E',500)]

 

# list
lst = ['A', 100, 'B', 200, 'C', 300, 'D', 400, 'E', 500]

# 방법1: zip 사용
new_lst1 = list(zip(lst[0::2], lst[1::2]))

# 방법2: for문과 list comprehension 사용
new_lst2 = [(lst[i],lst[i+1]) for i in range(0,len(lst),2)]

print(new_lst1)
print(new_lst2)

[('A', 100), ('B', 200), ('C', 300), ('D', 400), ('E', 500)]

 

마찬가지로 저는 zip를 활용하는 방법 1을 선호합니다. (코드가 짧을 수록 눈이 덜 피로한 것 같아서 ㅎㅎ...)

 

 

3. 두 개의 list에 있는 element 간에 짝을 지을 수 있는 모든 경우의 수

ex.  ['A', 'B', 'C']   [1, 2]  =>  [('A', 1), ('A', 2), ('B', 1), ('B', 2), ('C', 1), ('C', 2)]

 

lst1 = ['A','B','C']
lst2 = [1,2]

# python built-in library itertools 사용
from itertools import product
new_lst1 = list(product(lst1, lst2))

# list comprehension 사용
new_lst2 = [(x,y) for x in lst1 for y in lst2]

## 주의: lst1, lst2 순서를 바꾸면 결과가 달라짐
new_lst3 = [(x, y) for x in lst2 for y in lst1]

# 원하는 값
print(new_lst1)
print(new_lst2)

# 틀린 값
print(new_lst3)

[('A', 1), ('A', 2), ('B', 1), ('B', 2), ('C', 1), ('C', 2)]
[('A', 1), ('A', 2), ('B', 1), ('B', 2), ('C', 1), ('C', 2)]
[(1, 'A'), (1, 'B'), (1, 'C'), (2, 'A'), (2, 'B'), (2, 'C')]

 

아주 기본적인 내용이긴 하지만... 사람은 망각의 동물이기 때문에 가끔 씩 이렇게 정리를 하는 것이 기억력에 도움이 되는 것 같습니다 ㅎㅎ