以最小和为序的numpy数组

eyh26e7m  于 2022-11-24  发布在  其他
关注(0)|答案(1)|浏览(122)

给定以下numpy数组:

[[0 4]                     
   [5 0]]

我希望能够把他们在一个列表中的顺序最小的总和,但我希望它的工作一般,不仅是这两个数据,也为超过2。即,它也应该为这个工作:

[[0 4]                     
   [5 0]
   [1 2]]

因此,这将给予:

[[1 2] [0 4] [5 0]]

这是我得到:最小顺序= [] def最小总和(xy):总和= np.总和(xy,轴=1)

while sums >= 0:
   smallest_sum = np.amin(sums)
   #add smallest_sum to the list order_smallest
   #remove the smallest_sum from the list then?

我以为一个循环会工作,但我只是不能得到它的权利。

qcbq4gxm

qcbq4gxm1#

对要重新索引的和使用argsort

a =  np.array([[0, 4],
               [5, 0],
               [1, 2]])

out = a[np.argsort(a.sum(axis=1))]

输出:

array([[1, 2],
       [0, 4],
       [5, 0]])

相关问题