-
Notifications
You must be signed in to change notification settings - Fork 0
/
Flatten a List.py
32 lines (24 loc) · 1.46 KB
/
Flatten a List.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
"""
https://py.checkio.org/ru/mission/flatten-list/
уществует список, который содержит целые числа или другие вложенные списки,
которые могут содержать ещё несколько списков и целых чисел, которые затем... ну, вы поняли.
Вы должны поместить все целые значения в один плоский список. Порядок должен быть такой же,
как и в первоначальном списке, с представлением строки слева направо.
Мы должны скрыть эту программу от Николы, сделав её маленькой и лёгкой.
Поэтому ваш код должен быть короче, чем 140 символов (с пробелами) .
"""
def flat_list(array):
if not array:
return []
_list = []
if isinstance(array[0], int):
_list.append(array[0])
else:
_list.extend(flat_list(array[0]))
return _list + flat_list(array[1:])
if __name__ == '__main__':
assert flat_list([1, 2, 3]) == [1, 2, 3], "First"
assert flat_list([1, [2, 2, 2], 4]) == [1, 2, 2, 2, 4], "Second"
assert flat_list([[[2]], [4, [5, 6, [6], 6, 6, 6], 7]]) == [2, 4, 5, 6, 6, 6, 6, 6, 7], "Third"
assert flat_list([-1, [1, [-2], 1], -1]) == [-1, 1, -2, 1, -1], "Four"
print('Done! Check it')