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

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

1 格子状配列の生成
2 ソースコード
3 実行結果

格子状配列の生成

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

ソースコード

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

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

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

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

    print(xx)
    print(yy)

if __name__ == '__main__':
    main()

実行結果

サンプルプログラムの実行結果です。

[[1 2 3]
 [1 2 3]
 [1 2 3]
 [1 2 3]]

[[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

と同じ効果です。