把ndarray的一个元素取出变为整数
时间: 2024-05-11 11:19:10 浏览: 41
numpy.ndarray 实现对特定行或列取值
可以使用NumPy中的`astype()`函数将ndarray中的元素类型转换为整数类型,然后通过索引取出对应元素,例如:
```python
import numpy as np
arr = np.array([1.2, 2.5, 3.9])
# 将元素类型转换为整数类型
arr_int = arr.astype(int)
# 取出第一个元素并转换为整数类型
element_int = int(arr_int[0])
print(element_int) # 输出:1
```
在这个例子中,我们将一个包含小数的ndarray转换为整数类型,并取出了第一个元素并转换为整数类型。
阅读全文