numpy数据类型转换
使用 dtype 或者 astype 可以将numpy数组从类型X(如float)转换为类型Y(如uint8)
>>> A=np.array([ 1.2, 3., 5 ])
>>> A.dtype
dtype('float64')
#设置dtype
>>> A=np.array(A,dtype='uint8')
>>> A
array([1, 3, 5], dtype=uint8)
#使用astype函数
>>> A.astype("uint8")
array([1, 3, 5], dtype=uint8)
注意,当X类型的数据超出Y类型的表示范围时,会产生如下情况:
>>> B=np.array([1,255,256,257])
>>> B.astype('uint8')
array([ 1, 255, 0, 1], dtype=uint8)
>>> C=np.array([1,0,-1,-2])
>>> C.astype("uint8")
array([ 1, 0, 255, 254], dtype=uint8)
这与matlab有所不同, matlab的矩阵进行数据类型转换时,会将超出范围的值映射为其上下界值
>> B=[1, 256, -1];
>> B=uint8(B)
B =
1×3 uint8 行向量
1 255 0