[scala] Is there a way to take the first 1000 rows of a Spark Dataframe?