Skip to content

[SPARK-17177][SQL] Make grouping columns accessible from RelationalGroupedDataset #14742

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 4 commits into from
Closed

[SPARK-17177][SQL] Make grouping columns accessible from RelationalGroupedDataset #14742

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
29d8a5c
Fixed duplicated documentation problem + separated documentation for …
Jun 20, 2016
60491b9
merge with master
Jul 21, 2016
989df0e
Merge branch 'master' of https://github.com/apache/spark
Aug 21, 2016
02d6bab
Initial commit
Aug 21, 2016
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 10 additions & 0 deletions sql/core/src/main/scala/org/apache/spark/sql/RelationalGroupedDataset.scala
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -388,6 +388,16 @@ class RelationalGroupedDataset protected[sql](
pivot(pivotColumn, values.asScala)
}

/**
* Returns all grouping column names as an array.
*
* @since 2.1.0
*/
def columns: Array[String] = {
val groupingNamedExpressions = groupingExprs.map(alias)
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This might not be a column (a named expression). Thus, this might not be useful to end users.

groupingNamedExpressions.map(_.name).toArray
}

/**
* Applies the given serialized R function `func` to each group of data. For each unique group,
* the function will be passed the group key and an iterator that contains all of the elements in
Expand Down
3 changes: 3 additions & 0 deletions sql/core/src/test/scala/org/apache/spark/sql/DataFrameAggregateSuite.scala
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -85,6 +85,9 @@ class DataFrameAggregateSuite extends QueryTest with SharedSQLContext {
Row(new java.math.BigDecimal(3.0), new java.math.BigDecimal(3.0)),
Row(null, new java.math.BigDecimal(2.0)))
)

assert(df1.groupBy("key").columns.sameElements(Array("key")))
assert(df1.groupBy("value1", "value2").columns.sameElements(Array("value1", "value2")))
}

test("SPARK-17124 agg should be ordering preserving") {
Expand Down