python numpy矩阵的数据类型转换

在python中,numpy矩阵的数据类型转换需要调用astype(),直接修改dtype是无效的。调用astype返回数据类型修改后的矩阵,但是源数据的类型不会变,需要再对源数据的赋值进行操作才可以改变。例如

>>> a = np.array([1.0, 2.0])
>>> a.dtype
dtype('float64') # np默认的格式即为float64
>>> a.astype(np.int16)
array([1, 1], dtype=int16)
>>> a.dtype
dtype('float64') # 矩阵a的数据类型并没有变
>>> a = a.astype(np.int16) # 对源数据的赋值操作后a的数据类型变化
>>> a.dtype
dtype('int16')
>>> a
array([1, 1], dtype=int16)

若是直接修改dtype数据则会强制用新数据类型表示,并没有转换,就会出现输出错误数据的情况。

注意:在图像处理中,比如三通道的彩色图像矩阵,需要转换数据类型时一定要整个三通道矩阵一起转换,只转换一个通道的矩阵是不会起作用的。

你可能感兴趣的:(python numpy矩阵的数据类型转换)