Your code is almost correct.
Except two things :
i
is already used in yourfor
loop so don’t use it inval i =
- If you want to use the value of
i
in a string, use String Interpolation
So your code should look like :
for (i <- List ('a','b')) {
val df = sqlContext.sql(s"SELECT $i, col1, col2 FROM DF1")
df.show()
}
EDIT after author comment :
You can do this with a .map
and then a .reduceLeft
:
// All your dataframes
val dfs = Seq('a','b').map { i =>
sqlContext.sql(s"SELECT $i, col1, col2 FROM DF1")
}
// Then you can reduce your dataframes into one
val unionDF = dfs.reduceLeft((dfa, dfb) =>
dfa.unionAll(dfb)
)
2
solved Spark (Scala) execute dataframe within for loop