大宮盆栽デイズ - Omiya Bonsai Days -

冗談めかす埼玉のファインマン

@CheckiO@Flatten a List

この問題が属する、「Home」カテゴリがとてもむずかしい。これ以降、あまりにも難しかったら CheckiO から離脱して、なんらかの教本で勉強をしなおそうかな。

def flat_list(array):

    flatten = []
    for x in array:
        if type(x) is list:
            flatten.extend(flat_list(x))
        else:
            flatten.append(x)
    return flatten

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')