多次元のブロードキャストは？

2024年6月8日

多次元のブロードキャストは、異なる形状や次元の配列同士の演算を行う場合にも適用されます。NumPyは、ブロードキャストのルールに従って、配列の形状を自動的に拡張して演算を行います。

具体的な例を見てみましょう。

python
import numpy as np

# 3x3の行列
A = np.array([[1, 2, 3],
              [4, 5, 6],
              [7, 8, 9]])

# 1x3の行列
B = np.array([10, 20, 30])

# Aの各行にBを加える
result = A + B

print(result)

この場合、配列Aは3×3の行列で、配列Bは1×3の行列です。ブロードキャストのルールにより、配列BがAの形状に合わせて自動的に拡張され、各行に要素ごとの演算が行われます。

このようにして、多次元のブロードキャストを使用することで、異なる形状や次元の配列同士の演算を行うことができます。NumPyはブロードキャストを効果的に使用することで、コードの記述を簡潔にし、計算を効率化します。