NumPy: найдите k наименьших значений данного массива numpy

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

Напишите программу NumPy, чтобы найти k наименьших значений данного массива numpy.

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

Код Python:

import numpy as np
array1 = np.array([1, 7, 8, 2, 0.1, 3, 15, 2.5])
print("Original arrays:")
print(array1)
k = 4
result = np.argpartition(array1, k)
print("\nk smallest values:")
print(array1[result[:k]])

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

 Оригинальные массивы:
[1. 7. 8. 2. 0.1 3. 15. 2.5]
k наименьших значений:
[0.1 1. 2. 2.5]

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

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

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

Предыдущий: Напишите программу NumPy для перестановки столбцов заданного двумерного массива с использованием заданных позиций индекса.
Далее: NumPy Random Exercises Home.