• Home
• org.jetbrains.kotlinx
• ggdsl-dataframe
• 0.2.5-dev-1
• source code
• GroupedByWrapper.kt

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

org.jetbrains.kotlinx.ggdsl.dataframe.data.GroupedByWrapper.kt Maven / Gradle / Ivy

/*
* Copyright 2020-2022 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license.
*/

package org.jetbrains.kotlinx.ggdsl.dataframe.data

import org.jetbrains.kotlinx.dataframe.api.GroupBy
import org.jetbrains.kotlinx.dataframe.api.concat
import org.jetbrains.kotlinx.dataframe.api.map
import org.jetbrains.kotlinx.ggdsl.ir.data.CountedGroupedDataInterface

/**
 * Wrapper for a [GroupBy] implementing [CountedGroupedDataInterface].
 */
// TODO @Serializable
public data class GroupedByWrapper(public val groupBy: GroupBy): CountedGroupedDataInterface {
    override val keys: DataFrameWrapper
        get ()= DataFrameWrapper(groupBy.keys)
    override val groups: List
        get() = groupBy.groups.map { DataFrameWrapper(it) }.toList()

    override fun toLazy(): LazyGroupedDataFrame = LazyGroupedDataFrame(
        keys.df.columnNames(),
        DataFrameWrapper(groupBy.groups.concat())
    )
}