【学習ノート】匿名関数+リスト生成式+三元オペレータ、PythonでもC++スタイルが書ける?
902 ワード
今日文章を読んでいると、すごいように見えるものを見つけました.
この行のコードの役割は実際にリストを次元を下げることでありnumpyのflatten()の方法とはあまり差がない.
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]