numpy强制类型转换
时间: 2023-10-06 19:14:04 浏览: 66
强制类型转换
在numpy中,可以使用astype()函数来进行强制类型转换。astype()函数可以将数组转换为指定的数据类型。例如,可以将一个整型数组转换为浮点型数组或者布尔型数组。下面是astype()函数的用法示例:
```python
import numpy as np
# 定义一个整型数组
arr = np.array([1, 2, 3, 4, 5])
# 将整型数组转换为浮点型数组
arr_float = arr.astype(np.float)
# 将整型数组转换为布尔型数组
arr_bool = arr.astype(np.bool)
print(arr_float)
print(arr_bool)
```
输出结果如下:
```
[1. 2. 3. 4. 5.]
[ True True True True True]
```
在上面的示例中,我们首先定义了一个整型数组arr,然后使用astype()函数将它转换为浮点型数组arr_float和布尔型数组arr_bool。注意,astype()函数的参数是目标数据类型,可以使用numpy中定义好的数据类型,如np.float、np.int、np.bool等。
阅读全文