Pythonで2次元のnumpy配列を1次元の配列にフラット化する

2d numpy配列は、配列の配列です。この記事では、要素を1次元配列として取得するためにフラット化する方法を説明します。

平らにする

numpyのflatten関数は、2D配列を1D配列に変換する直接的な方法です。

例

import numpy as np
array2D = np.array([[31, 12, 43], [21, 9, 16], [0, 9, 0]])
# printing initial arrays
print("Given array:\n",array2D)
# Using flatten()
res = array2D.flatten()
# Result
print("Flattened array:\n ", res)

出力

上記のコードを実行すると、次の結果が得られます-

Given array:
[[31 12 43]
[21 9 16]
[ 0 9 0]]
Flattened array:
[31 12 43 21 9 16 0 9 0]

ラベリング付き

2D配列を1Dにフラット化するのと同様のことを行うravelと呼ばれる別の関数があります。

例

import numpy as np
array2D = np.array([[31, 12, 43], [21, 9, 16], [0, 9, 0]])
# printing initial arrays
print("Given array:\n",array2D)
# Using ravel
res = array2D.ravel()
# Result
print("Flattened array:\n ", res)

出力

上記のコードを実行すると、次の結果が得られます-

Given array:
[[31 12 43]
[21 9 16]
[ 0 9 0]]
Flattened array:
[31 12 43 21 9 16 0 9 0]