【学習ノート】匿名関数+リスト生成式+三元オペレータ、PythonでもC++スタイルが書ける?

902 ワード

今日文章を読んでいると、すごいように見えるものを見つけました.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]