Skip to content

Commit 80711c8

Browse files
committed
updating from master
1 parent cc41580 commit 80711c8

File tree

2 files changed

+14
-11
lines changed
  • core/src
    • main/kotlin/org/jetbrains/kotlinx/dataframe/columns
    • test/kotlin/org/jetbrains/kotlinx/dataframe/samples/api

2 files changed

+14
-11
lines changed

core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/columns/BaseColumn.kt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ package org.jetbrains.kotlinx.dataframe.columns
33
import org.jetbrains.kotlinx.dataframe.AnyBaseCol
44
import org.jetbrains.kotlinx.dataframe.AnyCol
55
import org.jetbrains.kotlinx.dataframe.AnyRow
6+
import org.jetbrains.kotlinx.dataframe.DataFrame
67
import org.jetbrains.kotlinx.dataframe.api.ColumnsSelectionDsl
78
import org.jetbrains.kotlinx.dataframe.impl.asList
89
import org.jetbrains.kotlinx.dataframe.impl.columnName

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

Lines changed: 13 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,8 @@
22

33
package org.jetbrains.kotlinx.dataframe.samples.api
44

5+
import org.jetbrains.kotlinx.dataframe.api.ColsAtAnyDepthColumnsSelectionDsl.colsAtAnyDepth
6+
import org.jetbrains.kotlinx.dataframe.api.FilterColumnsSelectionDsl.filter
57
import org.jetbrains.kotlinx.dataframe.api.add
68
import org.jetbrains.kotlinx.dataframe.api.after
79
import org.jetbrains.kotlinx.dataframe.api.chunked
@@ -590,7 +592,7 @@ class Access : TestBase() {
590592
df.select { name.allCols() }
591593

592594
// traversal of columns at any depth from here excluding ColumnGroups
593-
df.select { name.colsAtAnyDepth { !it.isColumnGroup() } }
595+
df.select { name.colsAtAnyDepth().filter { !it.isColumnGroup() } }
594596

595597
// SampleEnd
596598
}
@@ -626,7 +628,7 @@ class Access : TestBase() {
626628
df.select { Person::name.allCols() }
627629

628630
// traversal of columns at any depth from here excluding ColumnGroups
629-
df.select { Person::name.colsAtAnyDepth { !it.isColumnGroup() } }
631+
df.select { Person::name.colsAtAnyDepth().filter { !it.isColumnGroup() } }
630632
// SampleEnd
631633
}
632634

@@ -660,7 +662,7 @@ class Access : TestBase() {
660662
df.select { "name".allCols() }
661663

662664
// traversal of columns at any depth from here excluding ColumnGroups
663-
df.select { "name".colsAtAnyDepth { !it.isColumnGroup() } }
665+
df.select { "name".colsAtAnyDepth().filter { !it.isColumnGroup() } }
664666
// SampleEnd
665667
}
666668

@@ -719,13 +721,13 @@ class Access : TestBase() {
719721
}
720722

721723
// traversal of columns at any depth from here excluding ColumnGroups
722-
df.select { colsAtAnyDepth { !it.isColumnGroup() } }
724+
df.select { colsAtAnyDepth().filter { !it.isColumnGroup() } }
723725

724726
// traversal of columns at any depth from here including ColumnGroups
725727
df.select { colsAtAnyDepth() }
726728

727729
// traversal of columns at any depth with condition
728-
df.select { colsAtAnyDepth { it.name().contains(":") } }
730+
df.select { colsAtAnyDepth().filter() { it.name().contains(":") } }
729731

730732
// traversal of columns at any depth to find columns of given type
731733
df.select { colsAtAnyDepth().colsOf<String>() }
@@ -743,18 +745,18 @@ class Access : TestBase() {
743745
fun columnSelectorsModifySet() {
744746
// SampleStart
745747
// first/last n value- and frame columns in column set
746-
df.select { colsAtAnyDepth { !it.isColumnGroup() }.take(3) }
747-
df.select { colsAtAnyDepth { !it.isColumnGroup() }.takeLast(3) }
748+
df.select { colsAtAnyDepth().filter { !it.isColumnGroup() }.take(3) }
749+
df.select { colsAtAnyDepth().filter { !it.isColumnGroup() }.takeLast(3) }
748750

749751
// all except first/last n value- and frame columns in column set
750-
df.select { colsAtAnyDepth { !it.isColumnGroup() }.drop(3) }
751-
df.select { colsAtAnyDepth { !it.isColumnGroup() }.dropLast(3) }
752+
df.select { colsAtAnyDepth().filter { !it.isColumnGroup() }.drop(3) }
753+
df.select { colsAtAnyDepth().filter { !it.isColumnGroup() }.dropLast(3) }
752754

753755
// filter column set by condition
754-
df.select { colsAtAnyDepth { !it.isColumnGroup() }.filter { it.name().startsWith("year") } }
756+
df.select { colsAtAnyDepth().filter { !it.isColumnGroup() && it.name().startsWith("year") } }
755757

756758
// exclude columns from column set
757-
df.select { colsAtAnyDepth { !it.isColumnGroup() }.except { age } }
759+
df.select { colsAtAnyDepth().filter { !it.isColumnGroup() }.except { age } }
758760

759761
// keep only unique columns
760762
df.select { (colsOf<Int>() and age).distinct() }

0 commit comments

Comments
 (0)