TIL - 2차원 배열을 1차원 배열로

itertools

Python 내장 라이브러리 중 하나로, 효율적인 반복을 위한 함수

2차원 배열을 1차원 배열로 바꿔보자
	

실습

[소스코드]

import itertools, numpy as np

num_list = np.full((2, 3), 2)
print(num_list)

new_list = list(itertools.chain.from_iterable(num_list))
print(new_list)

new_list = list(itertools.chain(*num_list))
print(new_list)