0

I want to create a row (for any given k) such -

for k =2, graph will be [Row(1,2), Row(3,4)]

for k =3, graph will be [Row(1,2,3), Row(4,5,6), Row(7,8,9)]

I am new to scala and dont know how exactly can I insert values in row like this.

 import org.apache.spark.sql.Row    
 var graph = ArrayBuffer[Row]()
    
    val k = 3
    val k2 = k * k
    
    for (a <- 1 to k2) {
        graph += Row(a)
      }
MetalRaptor11
  • 73
  • 1
  • 1
  • 6

1 Answers1

0
val k = 3
val range = ArrayBuffer.range(1, k*k + 1) 
val rows = range.map(v => Row(v))
val grouped = rows.grouped(k).toBuffer
Matt
  • 620
  • 4
  • 11