• Home
• org.jetbrains.kotlinx
• ggdsl-dataframe
• 0.2.5-dev-1
• source code
• GroupedDataFrameContext.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.context.GroupedDataFrameContext.kt Maven / Gradle / Ivy

package org.jetbrains.kotlinx.ggdsl.dataframe.context

import org.jetbrains.kotlinx.dataframe.DataFrame
import org.jetbrains.kotlinx.dataframe.api.GroupBy
import org.jetbrains.kotlinx.dataframe.api.concat
import org.jetbrains.kotlinx.dataframe.api.first
import org.jetbrains.kotlinx.ggdsl.dataframe.data.DataFrameWrapper
import org.jetbrains.kotlinx.ggdsl.dataframe.data.LazyGroupedDataFrame

/**
 * Wrapper of [GroupBy] that allows to access grouping keys and dataframe columns.
 *
 * @property groupKey grouping keys accessor.
 * @property column columns accessor.
 */
public class GroupedDataFrameContext(private val groupBy: GroupBy)  {
    public val groupKey: DataFrame = groupBy.keys
    public val column: DataFrame = groupBy.groups.first() // TODO

    @PublishedApi
    internal fun toLazy(): LazyGroupedDataFrame = LazyGroupedDataFrame(
        groupKey.columns().map { it.name() },
        DataFrameWrapper(groupBy.groups.concat())
    )
}