Spark Python Limit Rows