Python - 2차원 리스트를 1차원 리스트로 만들기

JS · 02 Jan 2021

pythonbasic

다음과 같이 2차원 리스트를 1차원 리스트로 만드는 방법을 소개합니다.

list = [[1, 10], [2, 22], [3, 19], [4, 7]]
result = [1, 10, 2, 22, 3, 19, 4, 7]
  • sum()을 이용한 방법
  • list comprehension을 이용한 방법
  • itertools를 이용한 방법
  • numpy를 이용한 방법
  • reduce를 이용한 방법

sum()을 이용한 방법

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]

itertools를 이용한 방법

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]

numpy를 이용한 방법

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]

reduce를 이용한 방법

다음과 같이 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]

list comprehension을 이용한 방법

다음과 같이 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]
python
댓글을 보거나 쓰려면 이 버튼을 눌러주세요.
codechachaCopyright ©2019 codechacha