Merge pull request #1463 from CarloMariaProietti/from_rows_to_DataFrame

From rows to data frame

Author: Jolanrensen

core/api/core.api

@@ -4467,6 +4467,7 @@ public final class org/jetbrains/kotlinx/dataframe/api/ToDataFrameKt {
 	public static final fun toDataFrameColumnPathAnyNullable (Ljava/lang/Iterable;)Lorg/jetbrains/kotlinx/dataframe/DataFrame;
 	public static final fun toDataFrameColumnPathAnyNullable (Ljava/util/Map;)Lorg/jetbrains/kotlinx/dataframe/DataFrame;
 	public static final fun toDataFrameFromPairs (Ljava/lang/Iterable;)Lorg/jetbrains/kotlinx/dataframe/DataFrame;
+	public static final fun toDataFrameMapStringAnyNullable (Ljava/lang/Iterable;)Lorg/jetbrains/kotlinx/dataframe/DataFrame;
 	public static final fun toDataFramePairColumnPathAnyCol (Ljava/lang/Iterable;)Lorg/jetbrains/kotlinx/dataframe/DataFrame;
 	public static final fun toDataFrameT (Ljava/lang/Iterable;)Lorg/jetbrains/kotlinx/dataframe/DataFrame;
 }

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

@@ -72,6 +72,9 @@ public fun <T> Iterable<DataRow<T>>.toDataFrame(): DataFrame<T> {
     }
 }
 
+@JvmName("toDataFrameMapStringAnyNullable")
+public fun Iterable<Map<String, *>>.toDataFrame(): DataFrame<*> = map { it.toDataRow() }.toDataFrame()
+
 @JvmName("toDataFrameAnyColumn")
 public fun Iterable<AnyBaseCol>.toDataFrame(): AnyFrame = dataFrameOf(this)
 

core/src/test/kotlin/org/jetbrains/kotlinx/dataframe/api/toDataFrame.kt

@@ -425,6 +425,16 @@ class CreateDataFrameTests {
         df["value"].toList() shouldBe maps
     }
 
+    @Test
+    fun `should convert iterables of maps representing rows to DataFrame with value columns`() {
+        val maps: Iterable<Map<String, *>> = listOf(mapOf("a" to 1, "b" to true), mapOf("c" to 2, "d" to false))
+        val df = maps.toDataFrame()
+        df["a"][0] shouldBe 1
+        df["b"][0] shouldBe true
+        df["c"][1] shouldBe 2
+        df["d"][1] shouldBe false
+    }
+
     class NoPublicPropsClass(private val a: Int, private val b: String)
 
     @Test