Метод sort

Метод sort класса ndarray сортирует элементы массива вдоль указанной оси. Метод изменяет исходный массив, не создавая копию, что позволяет экономить память. По умолчанию сортировка выполняется по последней оси массива. Метод поддерживает сортировку как числовых, так и строковых данных. Для получения отсортированной копии массива используйте функцию np.sort.

Синтаксис

arr.sort([axis])

Параметр axis определяет ось, вдоль которой выполняется сортировка. Если параметр не указан или равен -1, сортировка выполняется по последней оси. Для одномерного массива сортировка выполняется по единственной оси. Для двумерного массива параметр может принимать значения 0 (сортировка по столбцам) или 1 (сортировка по строкам).

Пример

Давайте создадим одномерный массив с числами в случайном порядке и отсортируем его:

import numpy as np

arr = np.array([5, 2, 8, 1, 9, 3])
print(arr)
arr.sort()
print(arr)

Результат выполнения кода:

[5 2 8 1 9 3]
[1 2 3 5 8 9]

Пример

Отсортируем массив строк в алфавитном порядке:

import numpy as np

arr = np.array(['banana', 'apple', 'cherry', 'date'])
print(arr)
arr.sort()
print(arr)

Результат выполнения кода:

['banana' 'apple' 'cherry' 'date']
['apple' 'banana' 'cherry' 'date']

Пример

Отсортируем двумерный массив вдоль строк (по оси 1):

import numpy as np

arr = np.array([[3, 1, 2], [6, 4, 5], [9, 7, 8]])
print(arr)
arr.sort(axis=1)
print(arr)

Результат выполнения кода:

[[3 1 2]
 [6 4 5]
 [9 7 8]]
[[1 2 3]
 [4 5 6]
 [7 8 9]]

Пример

Отсортируем двумерный массив вдоль столбцов (по оси 0):

import numpy as np

arr = np.array([[3, 1, 2], [6, 4, 5], [9, 7, 8]])
print(arr)
arr.sort(axis=0)
print(arr)

Результат выполнения кода:

[[3 1 2]
 [6 4 5]
 [9 7 8]]
[[3 1 2]
 [6 4 5]
 [9 7 8]]

Пример

Создадим массив с числами с плавающей точкой и отсортируем его:

import numpy as np

arr = np.array([3.14, 2.71, 1.41, 1.73, 2.23])
print(arr)
arr.sort()
print(arr)

Результат выполнения кода:

[3.14 2.71 1.41 1.73 2.23]
[1.41 1.73 2.23 2.71 3.14]

Смотрите также

метод argsort,
который возвращает индексы отсортированных элементов
метод partition,
который частично сортирует массив
метод searchsorted,
который находит позицию для вставки элементов в отсортированный массив
метод argpartition,
который возвращает индексы частичной сортировки