@@ -10,6 +10,7 @@ import org.jetbrains.kotlinx.dataframe.Selector
1010import org.jetbrains.kotlinx.dataframe.annotations.AccessApiOverload
1111import org.jetbrains.kotlinx.dataframe.annotations.Interpretable
1212import org.jetbrains.kotlinx.dataframe.annotations.Refine
13+ import org.jetbrains.kotlinx.dataframe.columns.ColumnGroup
1314import org.jetbrains.kotlinx.dataframe.columns.ColumnReference
1415import org.jetbrains.kotlinx.dataframe.columns.FrameColumn
1516import 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+ */
7280public inline fun <T , R > DataFrame<T>.map (transform : RowExpression <T , R >): List <R > = rows().map { transform(it, it) }
7381
7482public inline fun <T , reified R > DataFrame<T>.mapToColumn (
0 commit comments