python列表生成式内必须定义匿名函数_【学习笔记】匿名函数+列表生成式+三元操作符，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]