Intellipaat Back

Explore Courses Blog Tutorials Interview Questions
0 votes
in Big Data Hadoop & Spark by (11.4k points)

When I use DataFrame groupby like this:


I will only get a DataFrame with columns "age" and "count(id)",but in df,there are many other columns like "name".

In all,I want to get the result as in MySQL,

"select name,age,count(id) from df group by age"

What should I do when use groupby in Spark?

1 Answer

0 votes
by (32.3k points)

Suppose you have a df that includes columns “name” and “age”, and on these two columns you want to perform groupBY.

Now, in order to get other columns also after doing a groupBy you can use join function.

chose_group = ['name', 'age']

data_counts = df.groupBy(chose_group).count().alias("counts")

data_joined = df.join(data_counts, chose_group).dropDuplicates()

Now, data_joined will have all columns including the count values.

Browse Categories