Numpyの次元の追加と入れ替え

画像をnumpy配列にしたい

Numpyの次元の追加と入れ替え

次元の追加

追加したいデータが入った変数[np.newaxis, :, :]

一次元追加

x.shapeが(3,5)だったら

x = x[np.newaxis, :, :]を入力すると

x.shapeは(1,3,5)になる

入れ替え

入れ替えたいデータが入った変数.transpose(2,0,1)みたいに入力

x.shapeが(1,3,5)のとき、1,3,5はインデックス0,1,2にそれぞれ対応している

なのでx = x.transpose(2,0,1)を実行すると

x.shapeは(5,1,3)になる。