scala-spark Матричное отображение

Question

У меня вопрос о отображении массива в scala. У меня есть следующий массив:

Array[(scala.collection.immutable.Set[String], com.trends.City, com.trends.State)] 

В принципе, я хочу, чтобы отобразить массив таким образом, что каждая строка в наборе будет иметь com.trends.City и государство прилагается к нему. Результат должен выглядеть примерно так:

Array[(String, com.trends.City, com.trends.State)] 

который, как flatMap, но я хочу com.trends быть там.

Я мог бы также преобразовать массив в RDD, если необходимо, и использовать flatMapValues, но я заинтересован в эффективности, может кто-нибудь сказать мне, что является лучшим способом?

Answer 1

Вы можете использовать flatMap на массиве лестницу, как это:

class City 
class State 
val array: Array[(scala.collection.immutable.Set[String], City, State)] = Array() 
array.flatMap(p => p._1.map(q => (q, p._2, p._3)))