Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- customer | Department
- ----------------------
- A | Food
- B | Home
- A | Office
- C | Home
- A | Home
- B | Office
- customer | Food | Home | Office
- -----------------------------------
- A 1 1 1
- B 0 1 1
- C 0 1 0
- val df = Seq(
- ("A", "Food"),
- ("B", "Home"),
- ("A", "Office"),
- ("C", "Home"),
- ("A", "Home"),
- ("B", "Office")
- ).toDF("customer", "department")
- df.groupBy("customer").pivot("department").agg(count("department"))
- .na.fill(0)
- +--------+----+----+------+
- |customer|Food|Home|Office|
- +--------+----+----+------+
- |B |0 |1 |1 |
- |C |0 |1 |0 |
- |A |1 |1 |1 |
- +--------+----+----+------+
Add Comment
Please, Sign In to add comment