簡単に言えば、numpy.newaxis
するために使用される寸法を増加することで、既存のアレイのを1つの以上のディメンションを使用する場合、一度。したがって、
1D配列は2D配列になります
2D配列は3D配列になります
3D配列は4D配列になります
4D配列は5D配列になります
等々..
これは、1Dアレイから2Dアレイへの昇格を示す視覚的なイラストです。
シナリオ1:np.newaxis
あなたがしたいときに便利になるかもしれない、明示的にどちらかに1次元配列に変換する行ベクトルまたは列ベクトル上の写真に示すように、。
例:
# 1D array
In [7]: arr = np.arange(4)
In [8]: arr.shape
Out[8]: (4,)
# make it as row vector by inserting an axis along first dimension
In [9]: row_vec = arr[np.newaxis, :] # arr[None, :]
In [10]: row_vec.shape
Out[10]: (1, 4)
# make it as column vector by inserting an axis along second dimension
In [11]: col_vec = arr[:, np.newaxis] # arr[:, None]
In [12]: col_vec.shape
Out[12]: (4, 1)
シナリオ2:いくつかの配列の追加を行うときなど、いくつかの操作の一部としてnumpyブロードキャストを利用したい場合。
例:
次の2つの配列を追加するとします。
x1 = np.array([1, 2, 3, 4, 5])
x2 = np.array([5, 4, 3])
これらを同じように追加しようとすると、NumPyは以下を発生させますValueError
。
ValueError: operands could not be broadcast together with shapes (5,) (3,)
この状況でnp.newaxis
は、NumPyがをブロードキャストできるように、を使用して配列の1つの次元を増やすことができます。
In [2]: x1_new = x1[:, np.newaxis] # x1[:, None]
# now, the shape of x1_new is (5, 1)
# array([[1],
# [2],
# [3],
# [4],
# [5]])
今、追加します:
In [3]: x1_new + x2
Out[3]:
array([[ 6, 5, 4],
[ 7, 6, 5],
[ 8, 7, 6],
[ 9, 8, 7],
[10, 9, 8]])
または、配列に新しい軸を追加することもできますx2
。
In [6]: x2_new = x2[:, np.newaxis] # x2[:, None]
In [7]: x2_new # shape is (3, 1)
Out[7]:
array([[5],
[4],
[3]])
今、追加します:
In [8]: x1 + x2_new
Out[8]:
array([[ 6, 7, 8, 9, 10],
[ 5, 6, 7, 8, 9],
[ 4, 5, 6, 7, 8]])
注:両方のケースで同じ結果が得られることを確認してください(ただし、一方は他方の転置です)。
シナリオ3:これはシナリオ1に似ています。ただし、np.newaxis
複数回使用して、配列をより高い次元に昇格させることができます。このような演算は、より高次の配列(つまり、テンソル)で必要になる場合があります。
例:
In [124]: arr = np.arange(5*5).reshape(5,5)
In [125]: arr.shape
Out[125]: (5, 5)
# promoting 2D array to a 5D array
In [126]: arr_5D = arr[np.newaxis, ..., np.newaxis, np.newaxis] # arr[None, ..., None, None]
In [127]: arr_5D.shape
Out[127]: (1, 5, 5, 1, 1)
np.newaxisとnp.reshapeの背景の詳細
newaxis
は、軸をマルチアレイに一時的に追加できる疑似インデックスとも呼ばれます。
np.newaxis
は、スライスオペレーターを使用して配列を再作成しながらnp.reshape
、配列を目的のレイアウトに再形成します(寸法が一致していることを前提とします。これは、が発生するために必要ですreshape
)。
例
In [13]: A = np.ones((3,4,5,6))
In [14]: B = np.ones((4,6))
In [15]: (A + B[:, np.newaxis, :]).shape # B[:, None, :]
Out[15]: (3, 4, 5, 6)
上記の例では、B
(ブロードキャストを使用するために)の最初の軸と2番目の軸の間に一時的な軸を挿入しました。欠落している軸は、ここでブロードキャスト操作を機能さnp.newaxis
せるために使用されます。
一般的なヒント:のNone
代わりに使用することもできnp.newaxis
ます。これらは実際には同じオブジェクトです。
In [13]: np.newaxis is None
Out[13]: True
PSまた、この素晴らしい答えを見てください:次元を追加するためのnewaxis vs reshape
except that it changes a row vector to a column vector?
最初の例は行ベクトルではありません。それはMATLABのコンセプトです。Pythonでは、行や列の概念がない1次元のベクトルです。行または列ベクトルは、第2の実施例と同様に、2- dimensonalある