【numpy】numpyのaxisについて

今回はnumpyのaxisについて解説します。

axis

まず、axisは「軸」という意味です。
axisによってnp.arrayのどの軸を対象としているかを決定します。

行列でいえば、ある処理を列に対してか、行に対して行うかを決定する時に用います。

以下の行列の総和を求めるnp.sumを用いて説明します。

axis=0 の時

axis=0の時には列に対して指定した処理を行います。

import numpy as np

arr = np.array([[1,2],
                [1,2]])

np.sum(arr,axis=0)
#=> array([2, 4])

図でいえば以下のようになります。

axis=1 の時

axis=1の時には行に対して、指定した処理を行います。

arr = np.array([[1,2],
                [1,2]])

np.sum(arr,axis=1)
#=> array([3, 3])

図でいえば以下のようになります。