Как преобразовать этот DataFrame в векторы? - PullRequest
0 голосов
/ 14 мая 2018

Здравствуйте, я новичок в Stack OverFlow, это мой первый вопрос.

Я работаю в Scala, и у меня есть это:

scala> val result = sql("Select col.pid, explode(col.tracks.track_uri) as track_uri From table")

scala> result.show(5)
+---+--------------------+
|pid|           track_uri|
+---+--------------------+
|  0|    0Uaiui7t32423WM5|
|  0|    6I97G45hbhJGkjhg|
|  0|    0Wqfhbf6fhbHJG34|
|  0|    1AW77b76KJVG7h6r|
|  0|    1lzkjd7HG8hjkjh9|
......
|45986|    k7sjHG8789jvgk|
+---+--------------------+

Я хочу преобразовать его ввот так:

pid     track1             track2              .....      trackN
0       0Uaiui7t32423WM5   6I97G45hbhJGkjhg    . . . 1lzkjd7HG8hjkjh9 
1          .
2          .
n       k7sjHG8789jvgk      .... .....                   ... 

Мне нужна помощь.Спасибо за преимущество.

1 Ответ

0 голосов
/ 14 мая 2018

Не explode

import org.apache.spark.sql.functions._

(0 until N).foldLeft(spark.table("table"))((df, i) => 
  df.withColumn(s"track$i", col("col.tracks.track_uri").getItem(i))
)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...