Python・NumPyで多項式をルジャンドル級数に変換する方法

はじめに

PythonのNumPyでは、numpy.polynomial.legendreモジュールに用意されたpoly2leg()メソッドを使うことで、多項式をルジャンドル級数へ簡単に変換できます。

このメソッドは、次数の低い項から高い項の順に並べられた多項式の係数を格納した配列を受け取り、それと等価なルジャンドル級数の係数（同様に低次から高次の順）を含む1次元配列を返します。引数には、多項式の係数を保持する1次元配列を指定します。

なお、ルジャンドル多項式は区間[-1, 1]における直交多項式の一種で、数値解析や近似理論の分野で広く利用されています。通常の冪級数形式の多項式をルジャンドル級数に変換すると、数値的に安定な計算や効率的な評価が可能になります。

変換の手順

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

まず、必要なライブラリをインポートします。

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

ステップ2：numpy.array()で配列を作成する

numpy.array()メソッドを使って、多項式の係数を表す配列を作成します。

c = np.array([1, 2, 3, 4, 5])

ここでは、[1, 2, 3, 4, 5]という係数を持つ多項式「1 + 2x + 3x² + 4x³ + 5x⁴」を扱います。

ステップ3：配列の情報を確認する

作成した配列を表示してみましょう。

print("Our Array...\n", c)

配列の次元数を確認します。

print("\nDimensions of our Array...\n", c.ndim)

データ型を確認します。

print("\nDatatype of our Array object...\n", c.dtype)

形状を確認します。

print("\nShape of our Array object...\n", c.shape)

ステップ4：poly2leg()でルジャンドル級数に変換する

多項式をルジャンドル級数に変換するには、legendre.poly2leg()メソッドを使用します。

print("\nResult (polynomial to legendre)...\n", L.poly2leg(c))

完全なサンプルコード

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

# numpy.array()メソッドで配列を作成
c = np.array([1, 2, 3, 4, 5])

# 配列を表示
print("Our Array...\n", c)

# 次元数を確認
print("\nDimensions of our Array...\n", c.ndim)

# データ型を確認
print("\nDatatype of our Array object...\n", c.dtype)

# 形状を確認
print("\nShape of our Array object...\n", c.shape)

# poly2leg()メソッドで多項式をルジャンドル級数に変換
print("\nResult (polynomial to legendre)...\n", L.poly2leg(c))

実行結果

Our Array...
    [1 2 3 4 5]

Dimensions of our Array...
1

Datatype of our Array object...
int64

Shape of our Array object...
(5,)

Result (polynomial to legendre)...
    [3. 4.4 4.85714286 1.6 1.14285714]

結果の解説

出力を見ると、入力配列は1次元・要素数5・データ型int64であることが確認できます。poly2leg()メソッドを実行すると、元の多項式と数学的に等価なルジャンドル級数の係数が浮動小数点型（float）の1次元配列として返されます。

つまり、多項式「1 + 2x + 3x² + 4x³ + 5x⁴」は、以下のルジャンドル級数と同一の関数を表していることになります。

3·P₀(x) + 4.4·P₁(x) + 4.857…·P₂(x) + 1.6·P₃(x) + 1.1428…·P₄(x)

ここでP₀〜P₄はそれぞれ0次から4次のルジャンドル多項式です。逆にルジャンドル級数から多項式へ戻したい場合は、leg2poly()メソッドを使用すれば変換できます。