按规范重新排序矢量列表

Question

我的数组A包含nvec 3D矢量列表A.shape==(nvec,3)。我想根据向量的规范重新排序A行：

norm=numpy.square(A).sum(axis=1)
rank=numpy.argsort(norm)

我试过A[rank,:]，但它给了我一个错误。

Answer 1

只需删除第二个下标：

A[rank]