np.argpartition — A Simple Illustrated Guide