Skip to content

Commit 70e180a

Browse files
map for column group fix
1 parent d95fc93 commit 70e180a

File tree

1 file changed

+9
-1
lines changed
  • core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api

1 file changed

+9
-1
lines changed

core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/map.kt

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@ import org.jetbrains.kotlinx.dataframe.Selector
1010
import org.jetbrains.kotlinx.dataframe.annotations.AccessApiOverload
1111
import org.jetbrains.kotlinx.dataframe.annotations.Interpretable
1212
import org.jetbrains.kotlinx.dataframe.annotations.Refine
13+
import org.jetbrains.kotlinx.dataframe.columns.ColumnGroup
1314
import org.jetbrains.kotlinx.dataframe.columns.ColumnReference
1415
import org.jetbrains.kotlinx.dataframe.columns.FrameColumn
1516
import org.jetbrains.kotlinx.dataframe.impl.columnName
@@ -25,7 +26,7 @@ import kotlin.reflect.typeOf
2526

2627
@Deprecated(DEPRECATED_ACCESS_API)
2728
@AccessApiOverload
28-
public inline fun <C, reified R> ColumnReference<C>.map(
29+
internal inline fun <C, reified R> ColumnReference<C>.map(
2930
infer: Infer = Infer.Nulls,
3031
noinline transform: (C) -> R,
3132
): ColumnReference<R> = createComputedColumnReference(name(), typeOf<R>(), infer) { transform(this@map()) }
@@ -69,6 +70,13 @@ public inline fun <T, R> DataColumn<T>.mapIndexed(
6970

7071
// region DataFrame
7172

73+
/**
74+
* Note: When this method is applied to a **[ColumnGroup]**,
75+
* its behavior differs from [DataColumn.map].
76+
* To apply `map` as if on a regular [DataColumn] (i.e., a column of [DataRow]s
77+
* whose values correspond to values in columns of the group),
78+
* call [ColumnGroup.asDataColumn] first.
79+
*/
7280
public inline fun <T, R> DataFrame<T>.map(transform: RowExpression<T, R>): List<R> = rows().map { transform(it, it) }
7381

7482
public inline fun <T, reified R> DataFrame<T>.mapToColumn(

0 commit comments

Comments
 (0)