Python : Depth 있는 List 를 단일 List 로 정리

#!/bin/env python
# coding: utf-8

#===========================================================================
# 수행시간 측정을 위한 코드
import timeit
start = timeit.default_timer()
#===========================================================================

 

1. 수집된 데이터들 중 Depth 가 생긴 List를 단일 List로 정리하는 코드

#===========================================================================
# Depth 있는 리스트를 단일 리스트로 변경
def flat_list(array):
	r_array = []

	def recursive_func(array):
		for i in array:
			r_array.append(i) if type(i) != list else recursive_func(i)

	recursive_func(array)
	return r_array
#==========================================================================

 

2. 테스트 코드

#===========================================================================
# 테스트 코드
if __name__ == '__main__':
	print(flat_list([1, 2, 3]))								# 예상 실행결과 :  [1, 2, 3]
	print(flat_list([1, [2, 2, 2], 4]))						# 예상 실행결과 :  [1, 2, 2, 2, 4]
	print(flat_list([[[2]], [4, [5, 6, [6], 6, 6, 6], 7]]))	# 예상 실행결과 :  [2, 4, 5, 6, 6, 6, 6, 6, 7]
	print(flat_list([-1, [1, [-2], 1], -1]))				# 예상 실행결과 :  [-1, 1, -2, 1, -1]
#==========================================================================

#===========================================================================
# 수행시간 측정을 위한 코드
stop = timeit.default_timer()
print(f"\n수행시간 : {stop - start:0.6f}")
#===========================================================================

 

4. 실행 결과