NumPy 配列に次元を追加する

1. numpy.expand_dims() 関数を使用して NumPy 配列に次元を追加する
2. Python の numpy.newaxis 関数を使用して NumPy 配列に次元を追加する

このチュートリアルでは、Python で NumPy 配列に新しい次元を追加する方法を紹介します。

numpy.expand_dims() 関数を使用して NumPy 配列に次元を追加する

numpy.expand_dims() 関数は、NumPy 配列に新しい次元を追加します。展開される配列と新しい軸を引数として取り、追加の次元を持つ新しい配列を返します。numpy.expand_dims() 関数の axis パラメーター内で展開する軸を指定できます。次のコード例を参照してください。

import numpy as np

array = np.array([1, 2, 3])
print(array.shape)

array = np.expand_dims(array, axis=0)
print(array.shape)

array = np.append(array, [[4, 5, 6]], axis=0)
print(array)

出力：

(3,)
(1, 3)
[[1 2 3]
 [4 5 6]]

上記のコードでは、最初に np.array() 関数を使用して 1D 配列 array を作成し、array.shape プロパティを使用して array の形状を出力しました。次に、np.expand_dims(array, axis=0) 関数を使用して array を 2D 配列に変換し、array.shape プロパティを使用して array の新しい形状を出力しました。最後に、np.append() 関数を使用して配列に新しい要素を追加し、配列の要素を出力しました。

Python の numpy.newaxis 関数を使用して NumPy 配列に次元を追加する

前のアプローチはその仕事をし、今のところうまく機能します。唯一の問題は、以前の方法が非推奨になり、将来的に Python の新しいバージョンでは機能しない可能性があることです。numpy.newaxis メソッドを使用して、前のメソッドと同じ目標を達成することもできますが、コードと複雑さはさらに少なくなります。この方法を使用すると、Python 以降のバージョンでサポートされないことを心配する必要もありません。numpy.newaxis メソッドは、Python の配列に新しい次元を追加します。

import numpy as np

array = np.array([1, 2, 3])
print(array.shape)

array = array[np.newaxis]
print(array.shape)

array = np.append(array, [[4, 5, 6]], axis=0)
print(array)

出力：

(3,)
(1, 3)
[[1 2 3]
 [4 5 6]]

array[np.newaxis] メソッドを使用して array を 2D 配列に変換し、array.shape プロパティを使用して array の新しい形状を出力しました。最後に、np.append() 関数を使用して配列に新しい要素を追加し、配列の要素を出力しました。