numpy数组转元组输出带类型包装的数值问题求助
问题:NumPy 2.0.1中转numpy数组为元组时保留了numpy类型包装
我将Python元组转换为numpy数组,用四元数函数处理后再转回元组。其中start_pos是元组(0,0,0),转换后的元组会被加入vertices列表。
原代码如下:
current_pos = np.array(start_pos) current_quaternion = Quaternion.from_axis_angle(np.array([0,0,1]), math.radians(rotation)) vertices.append(tuple(current_pos))
这段代码在旧环境运行正常,tuple(current_pos)会输出原始数值,但在Python 3.12.4 + NumPy 2.0.1的新环境中,输出变成(np.int64(0), np.int64(0), np.int64(0))这类带类型包装的数值。
由于vertices需要写入文件,必须是原始数值(如0.4999238475781956),不能带np.float64()这类包装,当前实际得到的列表为:
[(np.int64(0), np.int64(0), np.int64(0)), (np.float64(0.0), np.float64(0.0), np.float64(1.0)), (np.float64(0.4999238475781956), np.float64(0.008726203218641754), np.float64(1.8660254037844388)), (np.float64(0.0), np.float64(0.0), np.float64(2.0))]
解决方案
方法1:逐个元素转换为Python原生类型
遍历numpy数组的每个元素,使用item()方法自动转换为对应原生类型(整数转int、浮点数转float),再组成元组:
vertices.append(tuple(x.item() for x in current_pos))
也可以根据需求手动指定类型,比如统一转成浮点数:
vertices.append(tuple(float(x) for x in current_pos))
方法2:用tolist()转原生列表再转元组
利用numpy数组的tolist()方法直接得到原生类型的列表,再转换为元组,这个方法简洁且适合多维数组场景:
vertices.append(tuple(current_pos.tolist()))
方法3:创建数组时指定原生兼容类型
在初始化numpy数组时,指定dtype为Python原生类型,避免后续处理中保留numpy包装类型(若后续四元数操作未修改类型则有效):
current_pos = np.array(start_pos, dtype=float) vertices.append(tuple(current_pos.tolist()))
内容的提问来源于stack exchange,提问作者Gregory Driscoll
相关产品推荐
相关产品推荐

