小ネタ：numpyのflattenのメカニズム

numpyのndarrayをflattenすることがよくあるのですが、データがどのような順番で並ぶのかをきちんと理解してみます。

(3, 3, 3)のndarrayを作成。どの位置にどの値が割り当てられているかわかるように、値に連番で付与する。

import numpy as np

x = np.array([[[ 0,  1,  2], [ 3,  4,  5], [ 6,  7,  8]], 
              [[ 9, 10, 11], [12, 13, 14], [15, 16, 17]], 
              [[18, 19, 20], [21, 22, 23], [24, 25, 26]],])
print(x.shape)

out

(3, 3, 3)

これをnp.flatten()すると、どのように一次元に配置されるかを見てみます。

x.flatten()

もっともネストが深いところを順番に取って、１次元に並べていることがわかりました！割と直感的ですね。

out

array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, 15, 16,
       17, 18, 19, 20, 21, 22, 23, 24, 25, 26])

短いですが以上です！