NumPy: сортировка по первой, последней оси массива

NumPy: Array Object Exercise-29 с решением

Напишите программу NumPy для сортировки по первой, последней оси массива.
Образец массива: [[2,5], [4,4]]

Пример решения : -

Код NumPy:

import numpy as np
a = np.array([[4, 6],[2, 1]])
print("Original array: ")
print(a)
print("Sort along the first axis: ")
x = np.sort(a, axis=0)
print(x)
print("Sort along the last axis: ")
y = np.sort(x, axis=1)
print(y)

Пример вывода:

 Ожидаемый результат:
Исходный массив:                                                                        
[[4 6]                                                                                 
 [2 1]]                                                                                
Сортировка по первой оси:                                                             
[[2 1]                                                                                 
 [4 6]]                                                                                
Сортировка по последней оси:                                                              
[[1 2]                                                                                 
 [4 6]] 

Редактор кода Python:

Есть другой способ решить это решение? Внесите свой код (и комментарии) через Disqus.

Предыдущий: Напишите программу NumPy для сравнения двух массивов, используя numpy.
Далее: Напишите программу NumPy для сортировки пар имени и фамилии, возвращающих их индексы. (сначала по фамилии, затем по имени).