面试时碰到的问题

当时面试时面试官会当面拿张白纸，在上面写些问题给你做，在纸上写代码，当时写得很乱

lst = [1,2,3,[2,5],[1,5,7,[3,5,6],60], 10,9]
# 面试时碰到的问题， 把lst扁平化，当时写的差不多，就是乱，回来一试就写出来了
def foo(lst):
    result = []
    for n in lst:
        if type(n) is list:
            result.extend(foo(n))
        else:
            result.append(n)
    return result

print(foo(lst))