blob: 06dea714ef9103b4356a81573130fa9dc3130881 [file]
import org.jetbrains.kotlinx.dataframe.*
import org.jetbrains.kotlinx.dataframe.annotations.*
import org.jetbrains.kotlinx.dataframe.api.*
import org.jetbrains.kotlinx.dataframe.io.*
fun box(): String {
val df = dataFrameOf("name", "age", "city", "weight")(
"Alice", 15, "London", 54,
"Bob", 45, "Dubai", 87,
"Charlie", 20, "Moscow", null,
"Charlie", 40, "Milan", null,
"Bob", 30, "Tokyo", 68,
"Alice", 20, null, 55,
"Charlie", 30, "Moscow", 90,
)
val df2 = df.move { name and city }.under("nameAndCity")
val res = df2.convert { nameAndCity }.asFrame { it.remove { city } }
res.compareSchemas(strict = true)
return "OK"
}