小编Ala*_*ain的帖子

索引一维numpy.array作为矩阵

我试图在运行时索引具有不同维度的numpy.array.要检索例如*m数组的第一行a,您可以这样做

a[0,:]
Run Code Online (Sandbox Code Playgroud)

但是,如果恰好是1xn向量,则上面的代码会返回索引错误:

IndexError:索引太多了

由于代码需要尽可能高效地执行,我不想引入if声明.有没有人有一个方便的解决方案,理想情况下不涉及更改任何数据结构类型?

python numpy

7
推荐指数
1
解决办法
6831
查看次数

标签 统计

numpy ×1

python ×1