python - Numpy - 数组中行和列的自定义排序

Question

我可以根据存储在单独列表中的值对数组的行或列进行排序吗？

例如:

row_keys = [10, 11, 5, 6]
z = np.array([[2.77, 11., 4.1, 7.2],
                      [3.7, 2.2, 1.1, 0.5],
                      [2.5, 3.5, 5.0, 9.0],
                      [4.3, 2.2, 5.1, 6.1]])

应该产生类似的东西

array([[  2.5,   3.5,   5. ,   9. ],
       [  4.3,   2.2,   5.1,   6.1]
       [  2.77,  11. ,   4.1,   7.2],
       [  3.7,   2.2,   1.1,   0.5],
       ])

请将类似的功能应用于列。

Answer 1

行的另一种方式

z_rows = z[np.argsort(row_keys)]

对于列

z_columns = z.T[np.argsort(row_keys)].T