Python numpy 将二维数组转换为已排序的一维数组

Question

我有一个 2D numpy 数组，我想将该数组更改为已排序的 1D 数组。例如:

A = [[1,0,2],
     [0,3,0]]

我希望它是这样的:

B = [3,2,1,0,0,0]

知道如何使用 python 模块来做到这一点而不是编写排序算法或类似的东西吗？

谢谢

Answer 1

假设您希望按降序对它们进行排序 -

In [127]: A
Out[127]: [[1, 0, 2], [0, 3, 0]]

In [128]: B = np.sort(np.array(A).ravel())

In [129]: B[::-1]
Out[129]: array([3, 2, 1, 0, 0, 0])

基本上，它涉及三个步骤:使用 ravel() 压平数组，使用 np.sort 对其进行排序，然后反转索引以获得 的效果降序排序。