astype函数用于array中数值类型转换
Example
x = np.array([1, 2, 2.5])
x.astype(int)
输出
array([1, 2, 2])
Example
arr = np.arange((10))
print(arr, arr.dtype, sep="\n")
[0 1 2 3 4 5 6 7 8 9]
int32 #可以看到,他的数据类型为 int32
arr = arr.astype("float32")
print(arr, arr.dtype, sep="\n")
[0. 1. 2. 3. 4. 5. 6. 7. 8. 9.]
float32 #可以看到数据类型转换成了 float32