跳到主要内容

Numpy 数据类型

本小节将详述 Numpy 内置的数据类型，以及如何在创建数组对象时进行灵活指定、如何查看创建好的数组类型、以及如何更改数据类型。

1. 常见数据类型​

Numpy 支持的数据类型比 Python 内置的要多很多，而这也是 Numpy 如此灵活和强大的原因之一。

例如对于整数，在 Numpy 中，根据整数的位数不同所需要占据的空间大小不同，又对整数类型进行类细分，常见地可以分为 int8、int16、int32、int64 等等。

Numpy 支持的常用数据类型如下：

类型说明
int8、uint8分别表示有符号和无符号的8位整型，可表示的整数范围为-128 ~ 127、0 ~ 255
int16、uint16分别表示有符号和无符号的16位整型，可表示的整数范围为-32768 ~ 32767、0 ~ 65535
int32、uint32分别表示有符号和无符号的32位整型，可表示的整数范围为-2147483648 ~ 2147483647、0 ~ 4294967295
int64、uint64分别表示有符号和无符号的64位整型，可表示的整数范围为-9223372036854775808 ~ 9223372036854775807、0 ~ 18446744073709551615
float16、float32、float64、float128分别表示半精度浮点数、单精度浮点、双精度浮点、扩展精度浮点数
complex64、complex128、complex256分别用两个32位、64位、128位的浮点数表示的复数
bool存储True和False值的布尔类型
ObjectPython对象类型
string_类型代号S，固定长度的字符串类型，每个字符1个字节。例如，如果需要创建一个长度为10的字符串，应使用S10。
unicode_类型代号U，固定长度的unicode类型，跟字符串的定义方式一样，例如（U8）

2. 定义数据类型​

dtype 是一个特殊的对象，在该对象中定义了 ndarray 的数据类型与数据大小。通常我们在创建 ndarray 的时候，可以显示地利用 dtype 定义数组的细节信息。

案例​

在创建数组对象的时候，通过给 dtype 赋值，显式地定义一个 int8 类型的数组。

import numpy as np

arr0 = np.array([[1,2,3], [4,5,6]], dtype=np.int8)
arr0

Out:
    array([[1, 2, 3],
            [4, 5, 6]], dtype=int8)

在创建数组对象的时候，通过给 dtype 赋值，显式地定义一个 float16 类型的数组

arr1 = np.array([[1.2, -2.4, 3.1], [-4, 5.9, -6.7]], dtype=np.float16)
arr1

Out:
    array([[ 1.2, -2.4,  3.1],
        [-4. ,  5.9, -6.7]], dtype=float16)

在创建数组对象的时候，通过给 dtype 赋值，显式地定义一个 S2 字符串类型的数组

arr2 = np.array(list('ABCD'), ndmin=2, dtype='S2')
arr2

Out:
    array([[b'A', b'B', b'C', b'D']], dtype='|S2')

在创建数组对象的时候，通过给 dtype 赋值，显式地定义布尔类型的数组

arr3 = np.array([[-1,0,-3], [4,-5,6]], dtype=np.bool)
arr3

Out:
    array([[ True, False,  True],
           [ True,  True,  True]])

3. 查询数据类型​

同样的，对于已经定义好数组，也可以通过调用 dtype 对象，查看其数据类型。

案例​

逐个查看上面定义的数组的数据类型：

arr0.dtype
Out:
    dtype('int8')

arr1.dtype
Out:
    dtype('float16')

arr2.dtype
Out:
    dtype('S2')

arr3.dtype
Out:
    dtype('bool')

4. 更改数据类型​

对于已经定义好的数组，也可以通过 ndarray 的 astype 方法对 dtype 对象进行修改。

案例​

将整数类型的数组转换成浮点数类型，并查看转换之后的结果：

float_arr0 = arr0.astype(np.float)
float_arr0.dtype

Out:
    dtype('float64')

可以看到，在我们不指定具体的浮点数精度的时候，Numpy 也会自动去推断一个合理的类型（当然，很多时候并不是最优的类型，因此在处理大数据集时，建议定义完整）。

如果把浮点数转换成整数，则小数部分将会被截断舍弃。

int_arr1 = arr1.astype(np.int8)
int_arr1

array([[ 1, -2,  3],
       [-4,  5, -6]], dtype=int8)

对于字符串类型的数组，在某些情况是可以与浮点数进行转换的，例如：

string_arr4 = np.array([['1.2', '2.8'], ['4.3', '5.7']], dtype=np.string_)
string_arr4.astype(np.float16)

Out:
    array([[1.2, 2.8],
           [4.3, 5.7]], dtype=float16)

需要注意的是，astype 并非全能，在某些情况下，如果转换过程失败，则会引发一个ValueError。例如：

string_arr4.astype(np.int16)

ValueError                                Traceback (most recent call last)
<ipython-input-25-787558dcde3c> in <module>
----> 1 string_arr4.astype(np.int16)

ValueError: invalid literal for int() with base 10: '1.2'

通过观察上面的案例，我们会发现，利用 astype 方法会创造出一个新的数组。确实 astype 方法并不能在原有的数组上进行修改，因此如果你想保存改变后的数组，务必记得给它赋个值。

5. 小结​

本小节详细讲述了 Numpy 内置的数据类型，以及如何定义、查询、更改数组的数据类型。需要注意的是，在利用 astype 方法更改数据类型的时候，并不是 100% 成功的。