博客

Python numpy.common_type()函数

numpy.common_type()函数用于确定数据类型的公共部分。 如果两个数组的类型不同，则返回的类型是uint

语法

numpy.common_type( array1[, array2[, ...]])

参数

• array1, array2,... – 用于进行数据类型比较的数组。

返回值

返回数据类型的公共部分

注意，如果两个数组的类型不同，则返回的类型是uint

示例

import numpy as np

a = np.array([1, 2, 3])
b = np.array([4.0, 5.0, 6.0])
c = np.array(["a", "b", "c"])

print(np.common_type(a, b)) # 返回float类型
print(np.common_type(b, c)) # 返回uint类型，默认返回类型

输出：

float64
uint8

以上示例演示了如何使用numpy.common_type()函数来确定两个数组的公共类型。

第一个示例中数组a和数组b的数据类型分别为int64和float64，因此公共类型为float64。

第二个示例中，数组b和c的数据类型分别为float64和string_，类型不同，因此返回默认类型uint8。

示例2

可以通过传递多个参数来确定多个数组的公共类型。下面是一个示例：

import numpy as np

a = np.array([1, 2, 3])
b = np.array([4.0, 5.0, 6.0])
c = np.array(["a", "b", "c"])
d = np.array([1+1j, 3+4j, 5+3j])

print(np.common_type(a, b, c)) # 返回uint8类型
print(np.common_type(a, b, c, d)) # 返回complex128类型

输出：

uint8
complex128

以上示例演示了如何使用numpy.common_type()函数来确定多个数组的公共类型。

在第一次示例中，数组a，b和c的数据类型分别为int64，float64和string_，因此公共类型为默认类型uint8。

在第二个示例中，数组a，b，c和d的数据类型分别为int64，float64，string_和complex128，因此公共类型为complex128。

希望以上内容能够帮助您更好地理解和使用Python numpy.common_type()函数。