Сортировка массива / повторного массива python по столбцу

У меня довольно простой вопрос о том, как отсортировать весь массив / повторный массив по заданному столбцу. Например, учитывая массив:

import numpy as np
data = np.array([[5,2], [4,1], [3,6]])

Я хотел бы отсортировать данные по первому возвращаемому столбцу:

array([[3,6], [4,1], [5,2]])