NumPy 基礎(ndarray・ブロードキャスト)

数値計算の土台、NumPy。ndarrayとブロードキャストを理解すれば一気に楽になります。

import numpy as np

# 基本作成
a = np.array([1, 2, 3, 4, 5])     # 1次元配列
b = np.array([[1, 2], [3, 4]])     # 2次元配列
z = np.zeros((3, 4))               # 3x4のゼロ行列
i = np.eye(3)                       # 3x3の単位行列
r = np.random.randn(100)            # 標準正規分布から100サンプル

# 形状確認
print(a.shape)    # (5,)
print(b.shape)    # (2, 2)
print(b.dtype)    # int64
print(b.ndim)     # 2 (次元数)

# 演算 (ベクトル化)
doubled = a * 2                     # [2, 4, 6, 8, 10]
added = a + np.array([10,10,10,10,10])  # [11,12,13,14,15]
squared = a ** 2                    # [1, 4, 9, 16, 25]

# 統計関数
mean = a.mean()         # 3.0
std = a.std()           # 標準偏差
max_val = a.max()       # 5
argmax = a.argmax()     # 4 (最大値のインデックス)

# ブロードキャスト: 形状の違う配列同士を自動で揃える
matrix = np.array([[1, 2, 3],
                   [4, 5, 6],
                   [7, 8, 9]])
row = np.array([10, 20, 30])     # shape: (3,)
result = matrix + row             # 各行に row が足される
# [[11, 22, 33],
#  [14, 25, 36],
#  [17, 28, 39]]

# 各列の標準化 (z変換)
X = np.random.randn(100, 5)
X_std = (X - X.mean(axis=0)) / X.std(axis=0)

確認クイズ

(3,4) の行列 X に対し X.mean(axis=0) の結果のshapeは何か。

1. (3,)
2. (4,)
3. (3,4)
4. スカラー

こたえを見る

正解: 2. (4,)

(4,)。 axis=0 で行方向を潰すので、列ごとの平均が得られて長さ4のベクトルになります。 (列が4つあるため)。