[
https://issues.apache.org/jira/browse/SPARK-19428?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=15853272#comment-15853272
]
Herman van Hovell commented on SPARK-19428:
-------------------------------------------
You could also use a window function:
{noformat}
import org.apache.spark.sql.expressions.Window
val df = spark.range(10000).select($"id" % 100 as "a", $"id" as "b")
df.withColumn("rn",
row_number().over(Window.partitionBy($"a").orderBy($"b"))).filter($"rn" < 10)
{noformat}
> Ability to select first row of groupby
> --------------------------------------
>
> Key: SPARK-19428
> URL: https://issues.apache.org/jira/browse/SPARK-19428
> Project: Spark
> Issue Type: Brainstorming
> Components: SQL
> Affects Versions: 2.1.0
> Reporter: Luke Miner
> Priority: Minor
>
> It would be nice to be able to select the first row from {{GroupedData}}.
> Pandas has something like this:
> {{df.groupby('group').first()}}
> It's especially handy if you can order the group as well.
--
This message was sent by Atlassian JIRA
(v6.3.15#6346)
---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscr...@spark.apache.org
For additional commands, e-mail: issues-h...@spark.apache.org
