Python・NumPyのfloat_power()関数でべき乗を高精度に計算する方法

PythonのNumPyライブラリには、配列の各要素をべき乗するためのfloat_power()メソッドが用意されています。この記事では、float_power()を使って、第1引数の配列（底）を第2引数の配列（指数）でべき乗する方法を解説します。

float_power()メソッドとは

float_power()は、x1の各要素を、対応する位置にあるx2の要素でべき乗した結果を返します。x1とx2がどちらもスカラーの場合、結果もスカラーになります。

主なパラメータ：

• x1： 底となる配列またはスカラー
• x2： 指数となる配列またはスカラー

x1とx2は、ブロードキャスト可能で同じ形状にできる必要があります。

power()との違い

float_power()は通常のpower()関数と似ていますが、重要な違いがあります。整数型やfloat16、float32の入力は、最低でもfloat64以上の精度を持つ浮動小数点数に昇格されるため、結果は常に不正確（inexact）になります。これは一見デメリットに思えますが、負の指数に対しても実用的な結果を返し、正の指数でもオーバーフローが発生しにくいという利点があります。

また、負の値を非整数の指数でべき乗するとnanが返されます。複素数の結果を得たい場合は、入力をcomplex型にキャストするか、dtypeにcomplexを指定してください。

実装手順

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

import numpy as np

底となる値を準備します。

x1 = range(6)

底を表示します。

print("The bases...\n", x1)

次に、指数となる値を準備します。

x2 = [1.0, 2.0, 3.0, 3.0, 2.0, 1.0]

指数を表示します。

print("\nThe exponents...\n", x2)

float_power()メソッドを使って、べき乗の結果を取得します。

print("\nResult...\n", np.float_power(x1, x2))

完全なコード例

import numpy as np

# 底となる値
x1 = range(6)

# 底を表示
print("The bases...\n", x1)

# 指数となる値
x2 = [1.0, 2.0, 3.0, 3.0, 2.0, 1.0]

# 指数を表示
print("\nThe exponents...\n", x2)

# float_power()メソッドでべき乗を計算
# x1の各要素をx2の対応する指数でべき乗した結果を返す
# x1とx2が両方スカラーの場合、結果もスカラーになる
print("\nResult...\n", np.float_power(x1, x2))

実行結果

The bases...
range(0, 6)

The exponents...
[1.0, 2.0, 3.0, 3.0, 2.0, 1.0]

Result...
[ 0.  1.  8. 27. 16.  5.]

出力を見ると、各底が対応する指数でべき乗されていることがわかります。例えば、2³ = 8、3³ = 27、4² = 16という具合です。このようにfloat_power()を使えば、高精度なべき乗計算を簡単に行うことができます。