Как выполнить сортировку данных в массиве NumPy?

Пользователь

от deion , в категории: Python , год назад

Как выполнить сортировку данных в массиве NumPy?

Facebook Vk Ok Twitter LinkedIn Telegram Whatsapp

2 ответа

Пользователь

от dan_farrell , год назад

@deion 

Вы можете выполнить сортировку данных в массиве NumPy с помощью функции np.sort(). Эта функция создает отсортированную копию входного массива.


Вот пример:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
import numpy as np

# Создаем массив NumPy
arr = np.array([3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5])

# Выполняем сортировку массива
sorted_arr = np.sort(arr)

# Выводим отсортированный массив
print(sorted_arr)


Этот код создаст массив arr и затем выполнит сортировку этого массива с помощью np.sort(). Результат будет сохранен в переменной sorted_arr, которая будет содержать отсортированную копию массива arr.


Если вы хотите отсортировать массив arr в порядке возрастания, вы можете также использовать метод sort() вместо функции np.sort(). Вот пример:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
import numpy as np

# Создаем массив NumPy
arr = np.array([3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5])

# Выполняем сортировку массива
arr.sort()

# Выводим отсортированный массив
print(arr)


Этот код также создаст массив arr и затем выполнит сортировку этого массива. Однако в отличие от np.sort(), метод sort() изменит исходный массив arr, отсортировав его в порядке возрастания.

Пользователь

от akeem , 2 месяца назад

@deion 

Также можно использовать функцию argsort(), чтобы получить индексы отсортированных элементов в массиве NumPy.


Пример:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
import numpy as np

# Создаем массив NumPy
arr = np.array([3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5])

# Получаем индексы отсортированных элементов
sorted_indices = np.argsort(arr)

# Индексы отсортированных элементов
print(sorted_indices)

# Отсортированный массив
sorted_arr = arr[sorted_indices]
print(sorted_arr)


Этот пример показывает, как сначала получить индексы отсортированных элементов с помощью argsort(), а затем использовать эти индексы, чтобы получить отсортированный массив по этим индексам.