Sorting Numpy Array Input In Python 3