Refactor DataFrame row limiting code

Replaced the usage of 'rows().take().toDataFrame()' with the more concise 'take()' method in multiple files.

Author: ermolenkodev

core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/impl/io/writeJson.kt

@@ -11,7 +11,7 @@ import org.jetbrains.kotlinx.dataframe.api.indices
 import org.jetbrains.kotlinx.dataframe.api.isList
 import org.jetbrains.kotlinx.dataframe.api.name
 import org.jetbrains.kotlinx.dataframe.api.rows
-import org.jetbrains.kotlinx.dataframe.api.toDataFrame
+import org.jetbrains.kotlinx.dataframe.api.take
 import org.jetbrains.kotlinx.dataframe.columns.ColumnGroup
 import org.jetbrains.kotlinx.dataframe.columns.ColumnKind
 import org.jetbrains.kotlinx.dataframe.columns.FrameColumn
@@ -68,7 +68,7 @@ internal fun KlaxonJson.encodeRowWithMetadata(
 
             is FrameColumn<*> -> {
                 val data = if (rowLimit == null) encodeFrameWithMetadata(col[index])
-                else encodeFrameWithMetadata(col[index].rows().take(rowLimit).toDataFrame(), rowLimit)
+                else encodeFrameWithMetadata(col[index].take(rowLimit), rowLimit)
                 obj(
                     DATA to data,
                     METADATA to obj(
@@ -205,7 +205,7 @@ internal fun KlaxonJson.encodeDataFrameWithMetadata(
             NCOL to frame.columnsCount()
         ),
         KOTLIN_DATAFRAME to encodeFrameWithMetadata(
-            frame.rows().take(rowLimit).toDataFrame(),
+            frame.take(rowLimit),
             rowLimit = nestedRowLimit
         ),
     )

core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/jupyter/JupyterHtmlRenderer.kt

@@ -1,8 +1,7 @@
 package org.jetbrains.kotlinx.dataframe.jupyter
 
 import com.beust.klaxon.json
-import org.jetbrains.kotlinx.dataframe.api.rows
-import org.jetbrains.kotlinx.dataframe.api.toDataFrame
+import org.jetbrains.kotlinx.dataframe.api.take
 import org.jetbrains.kotlinx.dataframe.impl.io.encodeFrame
 import org.jetbrains.kotlinx.dataframe.io.DataFrameHtmlData
 import org.jetbrains.kotlinx.dataframe.io.DisplayConfiguration
@@ -71,7 +70,7 @@ internal inline fun <reified T : Any> JupyterHtmlRenderer.render(
                         "nrow" to df.size.nrow,
                         "ncol" to df.size.ncol,
                         "columns" to df.columnNames(),
-                        "kotlin_dataframe" to encodeFrame(df.rows().take(limit).toDataFrame()),
+                        "kotlin_dataframe" to encodeFrame(df.take(limit)),
                     )
                 }.toJsonString()
             } else {

core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/impl/io/writeJson.kt

@@ -11,7 +11,7 @@ import org.jetbrains.kotlinx.dataframe.api.indices
 import org.jetbrains.kotlinx.dataframe.api.isList
 import org.jetbrains.kotlinx.dataframe.api.name
 import org.jetbrains.kotlinx.dataframe.api.rows
-import org.jetbrains.kotlinx.dataframe.api.toDataFrame
+import org.jetbrains.kotlinx.dataframe.api.take
 import org.jetbrains.kotlinx.dataframe.columns.ColumnGroup
 import org.jetbrains.kotlinx.dataframe.columns.ColumnKind
 import org.jetbrains.kotlinx.dataframe.columns.FrameColumn
@@ -68,7 +68,7 @@ internal fun KlaxonJson.encodeRowWithMetadata(
 
             is FrameColumn<*> -> {
                 val data = if (rowLimit == null) encodeFrameWithMetadata(col[index])
-                else encodeFrameWithMetadata(col[index].rows().take(rowLimit).toDataFrame(), rowLimit)
+                else encodeFrameWithMetadata(col[index].take(rowLimit), rowLimit)
                 obj(
                     DATA to data,
                     METADATA to obj(
@@ -205,7 +205,7 @@ internal fun KlaxonJson.encodeDataFrameWithMetadata(
             NCOL to frame.columnsCount()
         ),
         KOTLIN_DATAFRAME to encodeFrameWithMetadata(
-            frame.rows().take(rowLimit).toDataFrame(),
+            frame.take(rowLimit),
             rowLimit = nestedRowLimit
         ),
     )

core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/jupyter/JupyterHtmlRenderer.kt

@@ -1,8 +1,7 @@
 package org.jetbrains.kotlinx.dataframe.jupyter
 
 import com.beust.klaxon.json
-import org.jetbrains.kotlinx.dataframe.api.rows
-import org.jetbrains.kotlinx.dataframe.api.toDataFrame
+import org.jetbrains.kotlinx.dataframe.api.take
 import org.jetbrains.kotlinx.dataframe.impl.io.encodeFrame
 import org.jetbrains.kotlinx.dataframe.io.DataFrameHtmlData
 import org.jetbrains.kotlinx.dataframe.io.DisplayConfiguration
@@ -71,7 +70,7 @@ internal inline fun <reified T : Any> JupyterHtmlRenderer.render(
                         "nrow" to df.size.nrow,
                         "ncol" to df.size.ncol,
                         "columns" to df.columnNames(),
-                        "kotlin_dataframe" to encodeFrame(df.rows().take(limit).toDataFrame()),
+                        "kotlin_dataframe" to encodeFrame(df.take(limit)),
                     )
                 }.toJsonString()
             } else {