다음과 같이 2차원 리스트를 1차원 리스트로 만드는 방법을 소개합니다.
list = [[1, 10], [2, 22], [3, 19], [4, 7]]
result = [1, 10, 2, 22, 3, 19, 4, 7]
1. sum()으로 2차원 리스트를 1차원 리스트로 변환
sum(iterable, start)은 start에 iterable의 모든 데이터를 더하는 함수입니다.
아래 코드에서 sum(list1, [])
은 [] + [1, 10] + [2, 22] + [3, 19] + [4, 7]
의 연산을 수행하게 됩니다.
python에서 [1, 10] + [2, 22]
는 [1, 10, 2, 22]
가 되기 때문에, 결국 2차원 리스트가 1차원 리스트로 변환됩니다.
list1 = [[1, 10], [2, 22], [3, 19], [4, 7]]
list2 = sum(list1, [])
print(list2)
Output:
[1, 10, 2, 22, 3, 19, 4, 7]
2. itertools로 2차원 -> 1차원 리스트 변환
itertools를 이용하여 2차원 리스트를 1차원 리스트로 만들 수 있습니다.
itertools.chain()
는 인자로 전달되는 iterable의 데이터를 연결하여 리턴해 줍니다.
명심해야하는 것은 인자 앞에 *
를 붙여주어야 하는 것입니다.
import itertools
list1 = [[1, 10], [2, 22], [3, 19], [4, 7]]
list2 = list(itertools.chain(*list1))
print(list2)
Output:
[1, 10, 2, 22, 3, 19, 4, 7]
itertools.chain.from_iterable()
itertools.chain.from_iterable()
를 사용하면 *
를 입력하지 않고 바로 인자를 전달할 수 있습니다.
결과는 위의 코드와 동일합니다.
import itertools
list1 = [[1, 10], [2, 22], [3, 19], [4, 7]]
list2 = list(itertools.chain.from_iterable(list1))
print(list2)
Output:
[1, 10, 2, 22, 3, 19, 4, 7]
3. numpy로 2차원 -> 1차원 리스트 변환
numpy 라이브러리를 이용하여 2차원 리스트를 1차원 리스트로 만들 수도 있습니다.
concatenate()
은 2차원 리스트의 데이터를 연결하여 1차원 리스트로 만들어 줍니다.
import numpy as np
list1 = [[1, 10], [2, 22], [3, 19], [4, 7]]
list2 = np.concatenate(list1).tolist()
print(list2)
Output:
[1, 10, 2, 22, 3, 19, 4, 7]
다음과 같이 numpy array의 flatten()
을 사용하여 1차원 리스트로 만들 수도 있습니다.
import numpy as np
list1 = [[1, 10], [2, 22], [3, 19], [4, 7]]
list2 = np.array(list1).flatten().tolist()
print(list2)
Output:
[1, 10, 2, 22, 3, 19, 4, 7]
4. reduce()로 2차원 -> 1차원 리스트 변환
다음과 같이 reduce()를 이용하여 1차원 리스트로 만들 수 있습니다.
from functools import reduce
import operator
list1 = [[1, 10], [2, 22], [3, 19], [4, 7]]
list2 = list(reduce(operator.add, list1))
print(list2)
Output:
[1, 10, 2, 22, 3, 19, 4, 7]
5. List comprehension으로 2차원 -> 1차원 리스트 변환
다음과 같이 list comprehension을 이용하여 2차원 리스트를 1차원 리스트로 만들 수 있습니다.
list1 = [[1, 10], [2, 22], [3, 19], [4, 7]]
list2 = [data for inner_list in list1 for data in inner_list]
print(list2)
Output:
[1, 10, 2, 22, 3, 19, 4, 7]
위의 list comprehension은 아래의 for loop로 구현한 것과 거의 동일합니다. 2개의 for loop로 하나의 list에 모든 데이터를 추가하는 것입니다.
list2 = []
for inner_list in list1:
for data in inner_list:
list2.append(data)
print(list2)
Output:
[1, 10, 2, 22, 3, 19, 4, 7]
Related Posts
- Python - Yaml 파일 파싱하는 방법
- Python - 파일 내용 삭제
- Python - for문에서 리스트 순회 중 요소 값 제거
- Python - 두 리스트에서 공통 요소 값 찾기
- Python - 문자열 앞(뒤)에 0으로 채우기
- Python - 공백으로 문자열 분리
- Python - 중첩 리스트 평탄화(1차원 리스트 변환)
- Python - 16진수 문자열을 Int로 변환
- Python - 두 날짜, 시간 비교
- Python f-string으로 변수 이름, 값 쉽게 출력 (변수명 = )
- Python - nonlocal과 global 사용 방법
- Python 바다코끼리 연산자 := 알아보기
- Python - pip와 requirements.txt로 패키지 관리
- Python - 딕셔너리 보기 좋게 출력 (pprint)
- Python - Requests 사용 방법 (GET/POST/PUT/PATCH/DELETE)
- Python - 온라인 컴파일러 사이트 추천
- Python - os.walk()를 사용하여 디렉토리, 파일 탐색
- Python - 문자열 비교 방법
- Python - Text 파일 읽고 쓰는 방법 (read, write, append)
- Python - 리스트에서 첫번째, 마지막 요소 가져오는 방법
- Python - 두개의 리스트 하나로 합치기
- Python - 리스트의 마지막 요소 제거
- Python - 리스트의 첫번째 요소 제거
- Python 소수점 버림, 4가지 방법
- Python 코드 안에서 버전 확인 방법
- Python 소수점 반올림, round() 예제
- Python - 리스트 평균 구하기, 3가지 방법
- Python - bytes를 String으로 변환하는 방법
- Python - String을 bytes로 변환하는 방법
- Python 버전 확인 방법 (터미널, cmd 명령어)
- Python - 람다(Lambda) 함수 사용 방법
- Python - dict 정렬 (Key, Value로 sorting)
- Python - range() 사용 방법 및 예제
- Python - 리스트를 문자열로 변환
- Python - 문자를 숫자로 변환 (String to Integer, Float)