numpy.find_common_type()函数介绍

numpy.find_common_type()函数是NumPy中用于找到一组数据类型的最小公共数据类型的函数。它可以根据给定的数据类型集合推断出一个尽可能小的数据类型，以适用于所有这些数据类型。

语法

numpy.find_common_type(types, scalar)

参数说明：

• types：数据类型列表或元组。
• scalar：可选的ndarray标量值，用于确定最终返回类型的大小和符号。如果省略，则返回类似于C语言sizeof()函数的最小类型，可以适用于所有数据类型。

返回值：最小公共数据类型。

示例

import numpy as np

dtypes = [np.float32, np.int8, np.uint8]

common_dtype = np.find_common_type(dtypes, None)
print(common_dtype)
# 输出：float32

在此示例中，我们使用numpy.find_common_type()函数找到三种不同的数据类型的最小公共数据类型。最小公共数据类型是float32。

如果我们指定scalar参数，它将用于确定最终返回类型的大小和符号。

import numpy as np

dtypes = [np.int16, np.int32, np.float32]

common_dtype = np.find_common_type(dtypes, np.array([1]))
print(common_dtype)
# 输出：float32

common_dtype = np.find_common_type(dtypes, np.array([1,2]))
print(common_dtype)
# 输出：int32

在第一个示例中，我们使用numpy.find_common_type()函数找到三种不同的数据类型的最小公共数据类型。但是，我们指定了一个ndarray标量值，因此最小公共数据类型是float32。

在第二个示例中，我们指定了一个大小为2的ndarray标量值。现在最小公共数据类型为int32。