博客

Python numpy.common_type()函数介绍

numpy.common_type()函数是一个用于找到可以同时表示输入数组类型的最小通用类型的函数。通常用于在广播时确定输出数组的数据类型。该函数返回的是 dtype 对象，表示能够表示所有输入数组的最小通用 dtype 对象。

语法如下：

numpy.common_type(*arrays[, out])

参数说明：

• arrays: 支持序列类型的数据，包括列表，元组等；
• out: 输出结果放置的位置，当 out=None 时，结果将被广播到新建数组中；

返回值：

返回表示可以展示 arrays 里的所有数据类型的最小数据类型的 numpy.dtype 类型。

示例说明：

示例1. 找寻两个数组的最小通用类型

import numpy as np

a = np.array([2, 3, 4])
b = np.array([1.3, 2.7, 3.1])

c = np.common_type(a, b)
print(c)

输出结果：

float64

示例2. 找寻三个数组的最小通用类型

import numpy as np

a = np.array([1, 2, 3])
b = np.array([3.3, 4.4])
c = np.array(['a', 'b', 'c'])

d = np.common_type(a, b, c)
print(d)

输出结果：

|S3

在此示例中，由于存在字符串和数字，因此最小通用类型将是字符序列类型 |S3。

需要注意的是，使用该函数可能会造成数据类型的损失，因此需要根据实际情况谨慎使用。