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

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

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

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

Код Python:

import numpy as np
data_type = [('name', 'S15'), ('class', int), ('height', float)]
students_details = [('James', 5, 48.5), ('Nail', 6, 52.5),('Paul', 5, 42.10), ('Pit', 5, 40.11)]
# create a structured array
students = np.array(students_details, dtype=data_type)   
print("Original array:")
print(students)
print("Sort by height")
print(np.sort(students, order='height'))    

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

 Исходный массив:
[(b'James ', 5, 48.5) (b'Nail', 6, 52.5) (b'Paul ', 5, 42.1)
 (b'Pit ', 5, 40.11)]
Сортировать по высоте
[(b'Pit ', 5, 40.11) (b'Paul', 5, 42.1) (b'James ', 5, 48.5)
 (b'Nail ', 6, 52.5)]

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

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

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

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