Pythonでルジャンドル多項式の疑似ファンデルモンド行列を生成する方法：x・y複素数座標配列とlegvander2d()の使い方

PythonのNumPyライブラリを使用して、ルジャンドル多項式の疑似ファンデルモンド（Vandermonde）行列を生成するには、legendre.legvander2d()メソッドを使います。このメソッドは疑似ファンデルモンド行列を返します。返される行列の形状は x.shape + (deg + 1,) となり、最後のインデックスは対応するルジャンドル多項式の次数を表します。データ型（dtype）は、変換後のxと同じ型になります。

パラメータについて

x、y： 点の座標を表す配列です。すべて同じ形状である必要があります。要素に複素数が含まれる場合は complex128、含まれない場合は float64 に変換されます。スカラー値は1次元配列に変換されます。

deg： [x_deg, y_deg] の形式で最大次数を指定するリストです。

実行手順

ステップ1：必要なライブラリをインポートする

import numpy as np
from numpy.polynomial import legendre as L

ステップ2：点の座標配列を作成する

numpy.array() メソッドを使って、すべて同じ形状の点の座標配列を作成します。ここでは複素数の配列を用意します。

x = np.array([-2.+2.j, -1.+2.j])
y = np.array([1.+2.j, 2.+2.j])

ステップ3：配列の内容を表示する

print("Array1...\n",x)
print("\nArray2...\n",y)

ステップ4：データ型を確認する

print("\nArray1 datatype...\n",x.dtype)
print("\nArray2 datatype...\n",y.dtype)

ステップ5：両方の配列の次元を確認する

print("\nDimensions of Array1...\n",x.ndim)
print("\nDimensions of Array2...\n",y.ndim)

ステップ6：両方の配列の形状を確認する

print("\nShape of Array1...\n",x.shape)
print("\nShape of Array2...\n",y.shape)

ステップ7：疑似ファンデルモンド行列を生成する

legendre.legvander2d() メソッドを使って、ルジャンドル多項式の疑似ファンデルモンド行列を生成します。ここでは x方向の最大次数を2、y方向の最大次数を3として指定します。

x_deg, y_deg = 2, 3
print("\nResult...\n",L.legvander2d(x,y, [x_deg, y_deg]))

完全なサンプルコード

import numpy as np
from numpy.polynomial import legendre as L

# numpy.array() メソッドで、すべて同じ形状の点の座標配列を作成
x = np.array([-2.+2.j, -1.+2.j])
y = np.array([1.+2.j, 2.+2.j])

# 配列を表示
print("Array1...\n",x)
print("\nArray2...\n",y)

# データ型を表示
print("\nArray1 datatype...\n",x.dtype)
print("\nArray2 datatype...\n",y.dtype)

# 両方の配列の次元を確認
print("\nDimensions of Array1...\n",x.ndim)
print("\nDimensions of Array2...\n",y.ndim)

# 両方の配列の形状を確認
print("\nShape of Array1...\n",x.shape)
print("\nShape of Array2...\n",y.shape)

# legendre.legvander2d() メソッドでルジャンドル多項式の疑似ファンデルモンド行列を生成
x_deg, y_deg = 2, 3
print("\nResult...\n",L.legvander2d(x,y, [x_deg, y_deg]))

実行結果

Array1...
    [-2.+2.j -1.+2.j]

Array2...
    [1.+2.j 2.+2.j]

Array1 datatype...
complex128

Array2 datatype...
complex128

Dimensions of Array1...
1

Dimensions of Array2...
1

Shape of Array1...
(2,)

Shape of Array2...
(2,)

Result...
     [[  1.   +0.j   1.  +2.j  -5.   +6.j -29.   -8.j  -2.  +2.j  -6.   -2.j
       -2. -22.j 74. -42.j -0.5 -12.j  23.5 -13.j 74.5 +57.j -81.5 +352.j]
     [  1.   +0.j   2.   +2.j  -0.5 +12.j -43.  +37.j  -1.  +2.j  -6.  +2.j
      -23.5 -13.j -31. -123.j  -5.  -6.j   2.  -22.j 74.5 -57.j 437.  +73.j]]

まとめ

このように、NumPyの legendre.legvander2d() メソッドを使えば、複素数の座標点からでも簡単にルジャンドル多項式の疑似ファンデルモンド行列を生成できます。複素数が含まれる場合、dtypeは自動的に complex128 に変換されるため、ユーザーが明示的に型を指定する必要はありません。多項式フィッティングや数値解析などの場面で活用できる便利な機能です。