python

描述：编写代码，计算出水仙花数

【水仙花数是一个三位数，三位数各位的立方之和等于三位数本身】

#while
a = 100
ge = 0
shi = 0
bai = 0
while a<1000:
    bai = a//100
    shi = (a - bai*100)//10
    ge = a - bai*100 - shi*10
    if a == (ge**3 + shi**3 + bai**3):
        print("水仙花数："+str(a))
    a +=1

#for
ge = 0
shi = 0
bai = 0

for a in range(100,1000):
    bai = a // 100
    shi = (a - bai * 100) // 10
    ge = a - bai * 100 - shi * 10
    if a == (ge ** 3 + shi ** 3 + bai ** 3):
        print("水仙花数：" + str(a))

输出结果为：
水仙花数：153
水仙花数：370
水仙花数：371
水仙花数：407

def flat_list(array):
    a = []
    for i in array:
        if isinstance(i, list):#判断元素i是不是一个序列，如果是返回true，否则返回false
            a.extend(flat_list(i)) #扩展：总是接受list参数，并且将该列表中的所有元素全部添加到a中
        else:
            a.append(i) #追加：如果不是一个list(即是一个元素)，直接添加到a中
    return a
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')