Using Numpy Sort A Boolean Function