【NumPy】格子状配列の生成 (numpy.meshgrid)

この記事では、Python言語とNumPyを用いて、格子状配列を生成する方法をソースコード付きで解説します。

格子状配列の生成

Pythonの数値計算モジュールNumPyでは、numpy.meshgridで格子状配列を生成できます。

ソースコード

サンプルプログラムのソースコードです。

# -*- coding: utf-8 -*-
import numpy as np

x = np.array([1,2,3])
y = np.array([4,5,6,7])

xx, yy = np.meshgrid(x, y)

print(xx)
"""
[[1 2 3]
 [1 2 3]
 [1 2 3]
 [1 2 3]]
"""

print(yy)
"""
[[4 4 4]
 [5 5 5]
 [6 6 6]
 [7 7 7]]
"""

これは

xx = np.repeat(x, (len(y))
yy = np.repeat(y, (len(x)).T

と同じ効果です。