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

Question

I am using the randomSplitfunction to get a small amount of a dataframe to use in dev purposes and I end up just taking the first df that is returned by this fu

Answer 1

Limit is very simple, example limit first 50 rows

val df_subset = data.limit(50)