Python如何转换numpy矩阵的数据类型
在进行数组和矩阵运算的时候,我们需要用到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数据则会强制用新数据类型表示,并没有转换,就会出现输出错误数据的情况。
注意:在图像处理中,比如三通道的彩色图像矩阵,需要转换数据类型时一定要整个三通道矩阵一起转换,只转换一个通道的矩阵是不会起作用的。
通过上面的示例,我们可以看出直接修改dtype是行不通的,小伙伴们一定要记住哦。更多Python学习推荐:云海天Python教程网。
来源:PY学习网:原文地址:https://www.py.cn/article.html