취미생활/CheckIO

[checkIO]Home 07.Flatten a List

우주먼지의하루 2020. 2. 16. 13:34
728x90

Nicola likes to categorize all sorts of things. He categorized a series of numbers and as the result of his efforts, a simple sequence of numbers became a deeply-nested list. Sophia and Stephan don't really understand his organization and need to figure out what it all means. They need your help to understand Nikolas crazy list.

There is a list which contains integers or other nested lists which may contain yet more lists and integers which then… you get the idea. You should put all of the integer values into one flat list. The order should be as it was in the original list with string representation from left to right.

We need to hide this program from Nikola by keeping it small and easy to hide. Because of this, your code should be shorter than 140 characters (with whitespaces).

Input data: A nested list with integers.

Output data: The one-dimensional list with integers.

Example:

flat_list([1, 2, 3]) == [1, 2, 3]

flat_list([1, [2, 2, 2], 4]) == [1, 2, 2, 2, 4]

flat_list([[[2]], [4, [5, 6, [6], 6, 6, 6], 7]]) == [2, 4, 5, 6, 6, 6, 6, 6, 7]

flat_list([-1, [1, [-2], 1], -1]) == [-1, 1, -2, 1, -1]

* 리스트를 풀어야한다.

* 140자보다 짧아야한다.

 

 

 

- 풀이

 

def flat_list(array):
    flatten_array = []
    for elem in array:
        if isinstance(elem, list):
            flatten_array.extend(flat_list(elem))
        else:
            flatten_array.append(elem)
    return flatten_array

 

* python list에서 append()와 extend()의 차이점

출처

https://hashcode.co.kr/questions/23/%ED%8C%8C%EC%9D%B4%EC%8D%AC%EC%97%90-append%EC%99%80-extend%EC%9D%98-%EC%B0%A8%EC%9D%B4%EC%A0%

 

 

* isinstance 함수도 type을 알아볼 수 있다.

 

isinstance(1,int)
# 1이 int 형인가. 결과는 True

isinstance(mylist,list)
# mylist가 list형인가. 결과는 True

 

 

반응형