スポンサーリンク

[numpy] mashgrid関数で座標行列生成

mashgrid関数で座標行列生成

mashgrid関数の機能

mashgrid関数の機能としては、第1引数と第2引数に1次元のnumpy配列渡すことで、
全組み合わせの座標の行列を返します。

ソースコード

IN

import numpy as np
x1 = np.array([1, 2, 3])
y1 = np.array([4, 5])
x2, y2 = np.meshgrid(x1, y1)

OUT(x2)

array([[1, 2, 3],
       [1, 2, 3]])

OUT(y2)

array([[4, 4, 4],
       [5, 5, 5]])

解説

実行結果のイメージとしては、以下図のように、
x1とy1の全組み合わせで、x座標の配列、y座標の配列を返します。
行数は第1引数、列数は第2引数の配列の長さになります。