如何从单元素Numpy数组中提取单个整数值?
解决单元素NumPy数组提取数值的问题
嘿,我太懂这种小问题卡壳的感觉了——有时候越基础的操作反而容易让人摸不着头脑!针对你要从单元素NumPy数组里提取14的需求,给你几个简单靠谱的方法:
方法1:索引取值
如果你的数组是一维单元素数组(比如arr = np.array([14])),直接用索引[0]就能拿到数值:
import numpy as np arr = np.array([14]) num = arr[0] print(num) # 输出14
如果是0维标量数组(直接创建arr = np.array(14)),可以用空索引[()]来取值:
arr = np.array(14) num = arr[()] print(num) # 输出14
方法2:.item()方法(最推荐)
这个方法不管你的数组是一维还是0维,都能直接返回对应的Python标量类型,通用性拉满:
arr1 = np.array([14]) num1 = arr1.item() # 得到14 arr2 = np.array(14) num2 = arr2.item() # 同样得到14
方法3:转换为Python内置类型
你也可以直接把数组转换成int类型,适合确定数组里是整数的情况:
arr = np.array([14]) num = int(arr) print(num) # 输出14
简单总结一下,.item()是最省心的选择,不用纠结数组的维度,直接就能拿到你要的数值~
内容的提问来源于stack exchange,提问作者Kevin Dow
相关产品推荐
相关产品推荐

