博客

Python numpy.common_type()函数详解

common_type()函数概述

numpy.common_type()函数用于从给定的数组值中查找最小公共类型。最大的浮点类型是单精度浮点型，最小的是byte类型。

函数语法

numpy.common_type()函数的使用方法如下：

numpy.common_type(*arrays)

参数

• *arrays：要在其中查找公共类型的数组。多个不定长度的数组可以通过逗号分隔作为该函数的参数传递。

返回值

numpy.common_type()函数返回一个dtype对象，表示所有传入数组的最小公共类型。

示例1

下面是从两个不同类型的数组中找到公共类型的numpy.common_type()函数的示例用法。

import numpy as np

arr1 = np.array([1, 2, 3])
arr2 = np.array([1.1, 2.2, 3.3])

common_type = np.common_type(arr1, arr2)
print(common_type)

输出结果为：

float64

示例2

下面是从一个包含数组的元组中找到公共类型的numpy.common_type()函数的示例用法。

import numpy as np

arr1 = [1, 2, 3]
arr2 = [1.1, 2.2, 3.3]
arr3 = ['a', 'b', 'c']

common_type = np.common_type(arr1, arr2, arr3)
print(common_type)

输出结果为：

<U3

在这个例子中，函数返回了<U3作为元素的数据类型，因为字符串数组被视为unicode字符串，每个字符需要4字节，因此每个元素需要至少12字节（3个字符* 4字节/字符）。