Python NumPyのlegval()メソッドでブロードキャストされた点xにおけるルジャンドル級数を評価する方法

Python NumPyでは、polynomial.legendre.legval()メソッドを使用することで、指定した点xにおけるルジャンドル級数を簡単に評価できます。この記事では、各パラメータの意味と実際のコード例を通じて、使い方を詳しく解説します。

パラメータの説明

第1パラメータ： x

評価対象となる点です。リストやタプルが渡された場合は自動的にndarrayに変換され、それ以外の場合はスカラーとしてそのまま扱われます。いずれの場合も、xまたはその要素は、自身および係数配列cの要素との加算・乗算をサポートしている必要があります。

第2パラメータ: c

係数の配列です。次数nの項に対応する係数がc[n]に格納されるように並べられています。cが多次元の場合、残りのインデックスは複数の多項式を表します。特に2次元の場合、係数はcの列に格納されていると考えることができます。

第3パラメータ: tensor

Trueの場合、係数配列の形状は右側に1が追加されて拡張されます（拡張の回数はxの次元数と同じ）。スカラーの次元は0として扱われます。その結果、cのすべての係数列がxのすべての要素に対して評価されます。Falseの場合、xは評価時にcの列へブロードキャストされます。このキーワード引数は、cが多次元である場合に特に有用です。デフォルト値はTrueです。

実行手順

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

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

次に、多次元の係数配列を作成します。

c = np.arange(4).reshape(2,2)

作成した配列を表示します。

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)

最後に、legval()メソッドを使って点[1, 2]におけるルジャンドル級数を評価します。ここではtensor=Falseを指定し、xをcの列にブロードキャストしています。

print("\nResult...\n",L.legval([1,2],c, tensor = False))

完全なコード例

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

# 多次元の係数配列を作成
c = np.arange(4).reshape(2,2)

# 配列を表示
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)

# 点x=[1, 2]でルジャンドル級数を評価
print("\nResult...\n",L.legval([1,2],c, tensor = False))

出力結果

Our Array...
    [[0 1]
    [2 3]]

Dimensions of our Array...
2

Datatype of our Array object...
int64

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

Result...
    [2. 7.]

このように、tensor=Falseを指定することで、点x=[1, 2]が係数配列cの各列にブロードキャストされ、それぞれの点でのルジャンドル級数の評価値[2. 7.]が得られます。