NumPyのブロードキャストとは？配列演算の仕組みをわかりやすく解説

NumPyは「Numerical Python」の略で、多次元配列オブジェクトや、それらの配列を効率的に処理するためのさまざまなメソッドを提供するPythonのライブラリです。

NumPyを使うことで、配列に対して幅広い操作を行うことができます。また、SciPyやMatplotlibなどのパッケージと組み合わせて使われることも多く、特に「NumPy + Matplotlib」の組み合わせは、MATLABの代替手段として広く認知されています。NumPyはオープンソースのパッケージであり、誰でも自由に利用できます。

なお、標準のPythonディストリビューションにはNumPyは含まれていないため、別途インストールが必要です。インストールにはパッケージ管理ツール「pip」を使用します。

NumPyのインストール方法

Windows環境の場合は、コマンドラインで以下のコマンドを実行します。

pip install numpy

このコマンドを実行した後、Python環境にNumPyをインポートすれば、すぐに使い始めることができます。

ndarray（N次元配列）とは

NumPyパッケージの中で最も重要なオブジェクトが、n次元配列である「ndarray」です。ndarrayは同じ型の要素の集合を定義し、その中の値には0から始まるインデックス（添字）を使ってアクセスできます。

ndarray内のすべての要素は、メモリ上で同じサイズのブロックを占有します。各要素のデータ型は「dtype」関数で確認でき、配列スライシングを使えば特定の要素を取り出すことも可能です。取り出された要素は、配列スカラー型のオブジェクトとして表現されます。

ブロードキャスト（Broadcasting）とは

NumPyにおけるブロードキャストとは、算術演算の際に異なる形状（shape）を持つ配列同士を扱えるようにする機能のことです。通常、形状の異なる配列同士の演算はエラーになりそうですが、ブロードキャスト機能によってエラーが発生することなく、スムーズに処理が実行されます。

サンプルコード

import numpy as np
arr_1 = np.array([4, 6, 8, 0, 3])
arr_2 = np.array([11,3,7,78, 999])
print("最初のndarrayは ")
print(arr_1)
print("2番目のndarrayは ")
print(arr_2)
arr_3 = arr_1 * arr_2
print("結果の配列は ")
print(arr_3)

実行結果

最初のndarrayは
[4 6 8 0 3]
2番目のndarrayは
[ 11 3 7 78 999]
結果の配列は
[ 44 18 56 0 2997]

コードの解説

• 必要なライブラリをPython環境にインポートします。

• 数値を格納した2つのndarrayを定義します。

• それぞれの配列をコンソールに出力して確認します。

• 3つ目の配列として、最初の2つのndarrayの積（要素ごとの掛け算）を定義します。

• 計算結果の配列をコンソールに表示します。

この例では、同じ形状の配列同士の演算を行いましたが、ブロードキャストの真価が発揮されるのは、たとえば1次元配列とスカラー値、あるいは異なる次元の配列同士を演算する場合です。NumPyが自動的に小さい方の配列を拡張して形状を揃えるため、ループ処理を書かずとも簡潔で高速なコードを実現できます。