python将浮点型数组转为整型数组
时间: 2023-10-09 10:08:27 浏览: 177
你可以使用NumPy库中的astype()函数将浮点型数组转换为整型数组。astype()函数可以接受一个参数,表示将数组转换为指定类型的数组。例如,将浮点型数组arr转换为整型数组,可以使用以下代码:
``` python
import numpy as np
arr = np.array([1.2, 2.6, 3.5])
arr_int = arr.astype(int)
print(arr_int)
```
输出结果为:
```
[1 2 3]
```
在这个例子中,我们先定义了一个浮点型数组arr。然后使用astype()函数将其转换为整型数组arr_int。最后,我们输出了转换后的整型数组arr_int。
相关问题
调用astype()方法把所有数据类型都转为浮点型
`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)
```
这样就可以把所有数据类型都转换为浮点型了。注意在转换前需要确保数组中只包含数值类型的数据。
将numpy转为float
Numpy中的数据类型已经默认为float类型,如果需要将其他类型转换为float类型,可以使用astype()函数,例如:
``` python
import numpy as np
arr = np.array([1, 2, 3], dtype='int32')
arr_float = arr.astype('float32')
print(arr_float) # [1. 2. 3.]
```
上面的代码将一个整型数组转换为浮点型数组。astype()函数的参数可以是字符串,也可以是numpy的数据类型对象。
阅读全文