Numpy数组转置异常:三维数组调用transpose(1,2)报错
问题分析与解决
- 报错根源:你的
controls是三维数组(轴索引为0、1、2),但调用transpose(1,2)时只传入了两个轴参数。numpy的transpose函数要求传入的参数是所有轴的完整排列,参数数量必须和数组维度数完全匹配。三维数组必须传递3个轴的顺序,仅传两个会导致轴数量不匹配,触发axes don't match array错误。 - 正确实现:如果你的需求是保持第0轴不变,交换第1和第2轴的位置,应该使用:
controls.transpose(0, 2, 1) - 验证代码:
import numpy as np # 构造与你数据形状一致的测试数组 controls = np.random.rand(600, 20000, 2) # 执行正确转置 transposed_arr = controls.transpose(0, 2, 1) print(transposed_arr.shape) # 输出:(600, 2, 20000),符合预期 - 简化写法:若仅需交换最后两个轴,还可以用更简洁的
swapaxes方法,效果完全一致:controls.swapaxes(1, 2)
内容的提问来源于stack exchange,提问作者Gavin An
相关产品推荐
相关产品推荐

