Python 中 NumPy 的数据类型

数组类型之间的转换

NumPy 中有些数据类型是与 C 中的数据类型是类似的。

Numpy 的类型 C 的类型 描述
np.bool bool 存储为字节的布尔值(True或False)
np.byte signed char 平台定义
np.ubyte unsigned char 平台定义
np.short short 平台定义
np.ushort unsigned short 平台定义
np.intc int 平台定义
np.uintc unsigned int 平台定义
np.int_ long 平台定义
np.uint unsigned long 平台定义
np.longlong long long 平台定义
np.ulonglong unsigned long long 平台定义
np.half / np.float16   半精度浮点数:符号位,5位指数,10位尾数
np.single float 平台定义的单精度浮点数:通常为符号位,8位指数,23位尾数
np.double double 平台定义的双精度浮点数:通常为符号位,11位指数,52位尾数。
np.longdouble long double 平台定义的扩展精度浮点数
np.csingle float complex 复数,由两个单精度浮点数(实部和虚部)表示
np.cdouble double complex 复数,由两个双精度浮点数(实部和虚部)表示。
np.clongdouble long double complex 复数,由两个扩展精度浮点数(实部和虚部)表示。

其中依赖平台本身定义的数据类型有一组固定大小的别名:

Numpy 的类型 C 的类型 描述
np.int8 int8_t 字节(-128到127)
np.int16 int16_t 整数(-32768至32767)
np.int32 int32_t 整数(-2147483648至2147483647)
np.int64 int64_t 整数(-9223372036854775808至9223372036854775807)
np.uint8 uint8_t 无符号整数(0到255)
np.uint16 uint16_t 无符号整数(0到65535)
np.uint32 uint32_t 无符号整数(0到4294967295)
np.uint64 uint64_t 无符号整数(0到18446744073709551615)
np.intp intptr_t 用于索引的整数,通常与索引相同 ssize_t
np.uintp uintptr_t 整数大到足以容纳指针
np.float32 float  
np.float64 / np.float_ double 与内置 python float 的精度相匹配。
np.complex64 float complex 复数,由两个32位浮点数(实数和虚数组件)表示
np.complex128 / np.complex_ double complex 与内置python 复合体的精度相匹配。

NumPy 数值类型是 dtype(数据类型)对象的实例,每个对象都具有独特的特征。使用下述语句导入 NumPy

>>> import numpy as np

在 dtypes 可作为 np.bool_,np.float32 等等。

有 5 种基本数字类型表示布尔值(bool),整数(int),无符号整数(uint)浮点(浮点数)和复数。名称中带有数字的那些表示该类型的位大小(即,在内存中表示单个值需要多少位)。某些类型(例如 int 和 intp)具有不同的位,取决于平台(例如,32位与64位计算机)。

数据类型可以用作将 python 数据转换为数组标量的函数,将 python 数字序列转换为该类型的数组,或作为许多 numpy 函数或方法接受的 dtype 关键字的参数。如:

>>> import numpy as np
>>> x = np.float32(1.0)
>>> x
1.0
>>> y = np.int_([1,2,4])
>>> y
array([1, 2, 4])
>>> z = np.arange(3, dtype=np.uint8)
>>> z
array([0, 1, 2], dtype=uint8)

数组类型也可以通过字符代码引用,主要是为了保持与较旧的包(如 Numeric)的向后兼容性(并不建议)。有些文档可能仍然引用这些,例如:

>>> np.array([1, 2, 3], dtype='f')
array([ 1.,  2.,  3.], dtype=float32)

要转换数组的类型,请使用 .astype() 方法(建议)或类型本身作为函数。例如:

>>> z.astype(float)                 
array([  0.,  1.,  2.])
>>> np.int8(z)
array([0, 1, 2], dtype=int8)

注意,在上面,我们使用 Python 的 float 对象作为 dtype。NumPy 中 int 是指 np.int_,bool 意味着 np.bool_,这 float 是 np.float_和 complex 是 np.complex_。其他数据类型没有类似的 Python 等价物。

要确定数组的类型,可以查看 dtype 属性:

>>> z.dtype
dtype('uint8')

dtype 对象还包含有关类型的信息,例如其位宽和字节顺序。数据类型也可以间接用于查询类型的属性,例如它是否为整数:

>>> d = np.dtype(int)
>>> d
dtype('int32')

>>> np.issubdtype(d, np.integer)
True

>>> np.issubdtype(d, np.floating)
False

数组标量

NumPy 通常将数组元素作为数组标量返回(带有关联 dtype 的标量)。数组标量与 Python 标量不同,但在大多数情况下它们可以互换使用。有一些例外,例如当代码需要标量的非常特定的属性或者它特定地检查值是否是 Python 标量时。通常,存在的问题很容易被显式转换数组标量到 Python 标量,采用相应的 Python 类型的功能(例如,固定的 int,float,complex,str,unicode)。

使用数组标量的主要优点是它们保留了数组类型(Python 可能没有匹配的标量类型,例如 int16)。因此,使用数组标量可确保数组和标量之间的相同行为,无论值是否在数组内。NumPy 标量也有许多与数组相同的方法。

建议在较多使用 ndarray 对象的地方,使用统一的数据类型,避免数据之间繁琐显式转换。

溢出错误

当值需要比数据类型中的可用内存更多的内存时,NumPy 数值类型的固定大小可能会导致溢出错误。如:

>>> np.power(100, 8, dtype=np.int64)
10000000000000000
>>> np.power(100, 8, dtype=np.int32)
1874919424

NumPy 和 Python 整数类型的行为在整数溢出方面存在显着差异。与 NumPy 不同,Python 的大小 int 是灵活的。这意味着 Python 整数可以扩展以容纳任何整数并且不会溢出。

NumPy 分别提供 numpy.iinfo 和 numpy.finfo 验证 NumPy 整数和浮点值的最小值或最大值:

>>> np.iinfo(np.int) # Bounds of the default integer on this system.
iinfo(min=-9223372036854775808, max=9223372036854775807, dtype=int64)
>>> np.iinfo(np.int32) # Bounds of a 32-bit integer
iinfo(min=-2147483648, max=2147483647, dtype=int32)
>>> np.iinfo(np.int64) # Bounds of a 64-bit integer
iinfo(min=-9223372036854775808, max=9223372036854775807, dtype=int64)

如果 64 位整数仍然太小,则结果可能会转换为浮点数。浮点数提供了更大但不精确的可能值范围。

>>> np.power(100, 100, dtype=np.int64) # Incorrect even with 64-bit int
0
>>> np.power(100, 100, dtype=np.float64)
1e+200

扩展精度

Python 的浮点数通常是 64 位浮点数,几乎等同于 np.float64。但某些情况下,使用更精确的浮点数可能会很有用。这在 numpy 中是否可行取决于硬件和开发环境:具体地说,x86 机器提供 80 位精度的硬件浮点,虽然大多数 C 编译器提供这一点作为它们的 long double 类型,MSVC(Windows 构建的标准)使 long double 等同于 double (64位)。NumPy 使编译器的 long double 作为 np.longdouble 可用(而 np.clongdouble 用于复数)。

NumPy 不提供比 C 的 long double 更高精度的 dtype;特别是 128 位 IEEE 四精度数据类型不可用。

参考资料:

1. NumPy 官方文档:https://numpy.org/devdocs/

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章