np.newaxisってなに

2024年6月8日

np.newaxisは、Numpy配列の新しい次元を追加するための便利な方法です。Numpyの配列に新しい軸（次元）を追加することで、データを効果的に操作したり、他の配列とブロードキャストしたりすることができます。

具体的には、np.newaxisをインデックスとして使用することで、新しい次元を持つ新しい配列を生成します。これにより、元の配列の形状を変更せずに、新しい軸を追加できます。

以下は、np.newaxisの使用例です。

python
import numpy as np

arr = np.array([1, 2, 3, 4, 5])
print(arr.shape)  # 出力: (5,)

# 新しい軸を追加して2次元配列に変換する
arr_2d = arr[:, np.newaxis]
print(arr_2d.shape)  # 出力: (5, 1)

この例では、元の1次元配列arrに新しい軸を追加して、arr_2dという新しい2次元配列を生成しています。np.newaxisを使用することで、arrの形状を(5,)から(5, 1)に変更しました。

np.newaxisは、ブロードキャストや配列の形状の変更など、多くのNumpy操作で便利に使用されます。