欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

python numpy矩阵的数据类型转换

程序员文章站 2024-03-18 18:54:58
...

在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数据则会强制用新数据类型表示,并没有转换,就会出现输出错误数据的情况。

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

相关标签: 数据类型转换