PythonのNumPy scimathで入力値の累乗の結果を返す方法

Pythonで入力値を累乗した結果をscimathを使って取得するには、scimath.power()メソッドを使用します。このメソッドはxのp乗、つまりx**pの計算結果を返します。xとpがスカラーであれば出力もスカラーになり、そうでなければ配列が返されます。

特に重要な特徴として、入力値xに負の値が含まれている場合、出力は自動的に複素数領域に変換されます。これにより、通常のべき乗演算ではエラーやNaNが発生するケースでも、数学的に正しい結果を得ることができます。

パラメータについて

x: 累乗の対象となる入力値です。

p: xを何乗するかを指定する指数です。xに複数の値が含まれる場合、pはスカラーであるか、xと同じ数の要素を持つ必要があります。後者の場合、結果はx[0]**p[0]、x[1]**p[1]、... というように要素ごとに計算されます。

手順

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

import numpy as np

array()メソッドを使用してNumPy配列を作成します。

arr = np.array([2, 4, 8, 16, 32])

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

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

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

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

配列のデータ型を確認します。

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

配列の形状を確認します。

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

scimathを使って入力値の累乗の結果を返すには、np.emath.power()メソッドを使用します。

print("\nResult...\n",np.emath.power(arr, 2))

完全なコード例

import numpy as np

# array()メソッドを使用してNumPy配列を作成
arr = np.array([2, 4, 8, 16, 32])

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

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

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

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

# scimath.power()メソッドで累乗を計算
print("\nResult...\n",np.emath.power(arr, 2))

実行結果

Our Array...
[ 2 4 8 16 32]

Dimensions of our Array...
1

Datatype of our Array object...
int64

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

Result...
[ 4 16 64 256 1024]

上記の出力からわかるように、配列の各要素[2, 4, 8, 16, 32]がそれぞれ2乗され、[4, 16, 64, 256, 1024]という結果が得られました。なお、np.emath.power()はnp.lib.scimath.power()のエイリアスであり、どちらの名前でも同じ機能を利用できます。