スポンサーリンク

numpy配列の次元追加

numpy配列の次元追加

行方向の次元追加

[](各括弧)の行方向に、np.newaxisを指定することで次元を増やせます。

IN

import numpy as np
arr1 = np.array([1, 2, 3])
arr1[np.newaxis, :]

OUT

array([[1, 2, 3]])

列方向の次元追加

[](各括弧)の列方向に、np.newaxisを指定することで次元を増やせます。

IN

import numpy as np
arr1 = np.array([1, 2, 3])
arr1[:, np.newaxis]

OUT

array([[1],
       [2],
       [3]])