Skip to content

Commit ac3d738

Browse files
committed
updated generated sources, rebased on master
1 parent dece7a4 commit ac3d738

File tree

4 files changed

+43
-37
lines changed

4 files changed

+43
-37
lines changed

core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/Nulls.kt

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -151,3 +151,35 @@ public fun <T> DataColumn<T?>.dropNA(): DataColumn<T> =
151151
}
152152

153153
// endregion
154+
155+
// region dropNaNs
156+
157+
public fun <T> DataFrame<T>.dropNaNs(whereAllNaN: Boolean = false, selector: ColumnsSelector<T, *>): DataFrame<T> {
158+
val cols = this[selector]
159+
160+
return if (whereAllNaN) drop { cols.all { this[it].isNaN } }
161+
else drop { cols.any { this[it].isNaN } }
162+
}
163+
164+
public fun <T> DataFrame<T>.dropNaNs(vararg cols: KProperty<*>, whereAllNaN: Boolean = false): DataFrame<T> =
165+
dropNaNs(whereAllNaN) { cols.toColumns() }
166+
167+
public fun <T> DataFrame<T>.dropNaNs(vararg cols: String, whereAllNaN: Boolean = false): DataFrame<T> =
168+
dropNaNs(whereAllNaN) { cols.toColumns() }
169+
170+
public fun <T> DataFrame<T>.dropNaNs(vararg cols: AnyColumnReference, whereAllNaN: Boolean = false): DataFrame<T> =
171+
dropNaNs(whereAllNaN) { cols.toColumns() }
172+
173+
public fun <T> DataFrame<T>.dropNaNs(cols: Iterable<AnyColumnReference>, whereAllNaN: Boolean = false): DataFrame<T> =
174+
dropNaNs(whereAllNaN) { cols.toColumnSet() }
175+
176+
public fun <T> DataFrame<T>.dropNaNs(whereAllNaN: Boolean = false): DataFrame<T> =
177+
dropNaNs(whereAllNaN) { all() }
178+
179+
public fun <T> DataColumn<T>.dropNaNs(): DataColumn<T> =
180+
when (typeClass) {
181+
Double::class, Float::class -> filter { !it.isNaN }.cast()
182+
else -> this
183+
}
184+
185+
// endregion

core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/api/flatten.kt

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,14 +2,22 @@ package org.jetbrains.kotlinx.dataframe.api
22

33
import org.jetbrains.kotlinx.dataframe.ColumnsSelector
44
import org.jetbrains.kotlinx.dataframe.DataFrame
5+
import org.jetbrains.kotlinx.dataframe.columns.ColumnReference
56
import org.jetbrains.kotlinx.dataframe.impl.api.flattenImpl
7+
import org.jetbrains.kotlinx.dataframe.impl.columns.toColumns
8+
import kotlin.reflect.KProperty
69

710
// region DataFrame
811

912
public fun <T> DataFrame<T>.flatten(): DataFrame<T> = flatten { all() }
1013

11-
public fun <T, C> DataFrame<T>.flatten(
12-
columns: ColumnsSelector<T, C>
13-
): DataFrame<T> = flattenImpl(columns)
14+
public fun <T, C> DataFrame<T>.flatten(columns: ColumnsSelector<T, C>): DataFrame<T> = flattenImpl(columns)
15+
16+
public fun <T> DataFrame<T>.flatten(vararg columns: String): DataFrame<T> = flattenImpl { columns.toColumns() }
17+
18+
public fun <T, C> DataFrame<T>.flatten(vararg columns: KProperty<C>): DataFrame<T> = flattenImpl { columns.toColumns() }
19+
20+
public fun <T, C> DataFrame<T>.flatten(vararg columns: ColumnReference<C>): DataFrame<T> =
21+
flattenImpl { columns.toColumns() }
1422

1523
// endregion

core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/test/sampleExtensions.kt

Lines changed: 0 additions & 17 deletions
This file was deleted.

core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/test/usage.kt

Lines changed: 0 additions & 17 deletions
This file was deleted.

0 commit comments

Comments
 (0)