我试图在运行时索引具有不同维度的numpy.array.要检索例如*m数组的第一行a,您可以这样做
a
a[0,:]
但是,如果恰好是1xn向量,则上面的代码会返回索引错误:
IndexError:索引太多了
由于代码需要尽可能高效地执行,我不想引入if声明.有没有人有一个方便的解决方案,理想情况下不涉及更改任何数据结构类型?
if
python numpy
numpy ×1
python ×1