How to get the two smallest values from a numpy array

You can use numpy.partition to get the lowest k+1 items:

A, B = np.partition(x, 1)[0:2]  # k=1, so the first two are the smallest items

In Python 3.x you could also use:

A, B, *_ = np.partition(x, 1)

For example:

import numpy as np
x = np.array([5, 3, 1, 2, 6])
A, B = np.partition(x, 1)[0:2]
print(A)  # 1
print(B)  # 2

Related