pyspark sample fixed number of rows