NumPy: сортировка заданного массива формы 2 вдоль первой оси, последней оси и на плоском массиве

Сортировка и поиск в NumPy: упражнение-1 с решением

Напишите программу NumPy для сортировки заданного массива формы 2 вдоль первой оси, последней оси и на плоском массиве.

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

Код Python:

import numpy as np
a = np.array([[10,40],[30,20]])
print("Original array:")
print(a)
print("Sort the array along the first axis:")
print(np.sort(a, axis=0))
print("Sort the array along the last axis:")
print(np.sort(a))
print("Sort the flattened array:")
print(np.sort(a, axis=None))

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

 Исходный массив:
[[10 40]
 [30 20]]
Отсортируйте массив по первой оси:
[[10 20]
 [30 40]]
Отсортируйте массив по последней оси:
[[10 40]
 [20 30]]
Сортировать уплощенный массив:
[10 20 30 40]

Иллюстрированная презентация:

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

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

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