按索引对 numpy.array 行进行排序

【问题标题】：Sort numpy.array rows by indices按索引对 numpy.array 行进行排序
【发布时间】：2014-10-31 05:30:44
【问题描述】：

我有 2D numpy.array 和一组索引：

a = array([[0, 0], [0, 1], [1, 0], [1, 1]])
ix = (2, 0, 3, 1)

如何按索引对数组的行进行排序？预期结果：

array([[1, 0], [0, 0], [1, 1], [0, 1]])

我尝试使用numpy.take，但它只适用于一维数组。

【问题讨论】：

【解决方案1】：

您实际上可以为此使用ndarray.take()。诀窍是提供第二个参数 (axis)：

>>> a.take(ix, 0)
array([[1, 0],
       [0, 0],
       [1, 1],
       [0, 1]])

（如果没有axis，则在获取元素之前将数组展平。）

或者：

>>> a[ix, ...]
array([[1, 0],
       [0, 0],
       [1, 1],
       [0, 1]])

【讨论】：