Сортировать numpy.array строк по индексам

У меня 2Dnumpy.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, но это работает, как я ожидаю, только с 1D массивами.