【学习笔记】匿名函数+列表生成式+三元操作符，Python也能写出C++风格？

今天看文章的时候发现一个看起来好厉害的东西：flatten = lambda x: [y for l in x for y in flatten(l)] if type(x) is list else [x]

作为一个只会一点点Python的渣渣看见这种C++风格的代码简直给整懵了，然后研究了一会，参考了 Python中flatten用法 一文后，自己把这行代码的等价形式写了一遍，并且实验了一下，发现结果是一样的。

这行代码的作用实际上就是让列表降维，跟numpy里的flatten()方法差不多

# 等价形式
import numpy as np

flatten = lambda x: [y for l in x for y in flatten(l)] if type(x) is list else [x]

A = np.array([[1,2],[3,4],[5,6]])
B = [[1,2],[3,4],[5,6]]

print(flatten(A))
print(flatten(B))

def my_flatten(x):
    if type(x) is list:
        Y = []
        for l in x:
            for y in my_flatten(l):
                Y.append(y)
        return Y
    else:
        return [x]

print(my_flatten(A))
print(my_flatten(B))

# 输出
[array([[1, 2],
       [3, 4],
       [5, 6]])]
[1, 2, 3, 4, 5, 6]
[array([[1, 2],
        [3, 4],
        [5, 6]])]
[1, 2, 3, 4, 5, 6]