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]])