调用astype()方法把所有数据类型都转为浮点型
时间: 2023-12-07 14:05:41 浏览: 143
`astype()` 方法可以用于转换数组的数据类型,例如将整数型转换为浮点型。以下是一个示例:
``` python
import numpy as np
# 生成一个整数型数组
arr = np.array([1, 2, 3, 4, 5])
# 将整型数组转换为浮点型数组
arr = arr.astype(float)
print(arr)
```
输出结果为:
```
[1. 2. 3. 4. 5.]
```
在你的情境中,可以使用`astype()`方法将`X_train`和`X_test`转换为浮点型,示例如下:
``` python
X_train = X_train.astype(float)
X_test = X_test.astype(float)
```
这样就可以把所有数据类型都转换为浮点型了。注意在转换前需要确保数组中只包含数值类型的数据。
阅读全文