Python NumPyのcommon_type()メソッドで入力配列に共通するスカラー型を取得する方法

PythonのNumPyでは、numpy.common_type()メソッドを使うことで、複数の入力配列に共通するスカラー型を取得できます。第1引数には対象となる入力配列を指定します。

重要なポイントとして、戻り値は常に不正確な（浮動小数点）スカラー型になります。仮に入力された配列がすべて整数型であったとしても、結果は浮動小数点型です。また、入力のいずれかに整数配列が含まれる場合、返される最小精度の型は64ビット浮動小数点数（float64）のdtypeとなります。

さらに、int64とuint64を除くすべての入力配列は、情報を失うことなく返されたdtypeへ安全にキャストできるという特徴があります。

実行手順

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

import numpy as np

次に、入力配列に共通するスカラー型を取得するために、numpy.common_type()メソッドを呼び出します。

コード例

import numpy as np

# 入力配列に共通するスカラー型を返すには、
# Python NumPyのnumpy.common_type()メソッドを使用します。
# 第1引数には入力配列を指定します。

print("NumPyのcommon_type()メソッドを使用\n")
print("結果...",np.common_type(np.arange(3,dtype=np.float32)))
print("結果...",np.common_type(np.arange(3,dtype=np.float32), np.arange(2)))
print("結果...",np.common_type(np.arange(3), np.array([22, 2.j]), np.array([32.9])))
print("結果...",np.common_type(np.arange(3), np.array([22, 39]), np.array([32.9])))
print("結果...",np.common_type(np.arange(3,dtype=np.int32), np.arange(2)))

このコードでは、float32のみの配列、整数配列との組み合わせ、複素数を含む配列など、さまざまなパターンでcommon_type()を呼び出しています。

実行結果

NumPyのcommon_type()メソッドを使用

結果... <class 'numpy.float32'>
結果... <class 'numpy.float64'>
結果... <class 'numpy.complex128'>
結果... <class 'numpy.float64'>
結果... <class 'numpy.float64'>

結果の解説

• float32単体の場合 → float32がそのまま返される
• float32と整数配列の組み合わせ → float64に昇格される
• 複素数を含む配列がある場合 → complex128が返される
• 整数配列とfloat64の組み合わせ → 最小精度としてfloat64が返される

このように、common_type()メソッドは配列間の演算を行う前に、互換性のあるデータ型を事前に確認したい場合に非常に便利です。特に異なるdtypeの配列を混在させて扱う際に、意図しない型変換や精度低下を防ぐのに役立ちます。