2
2
3
3
package org.jetbrains.kotlinx.dataframe.samples.api
4
4
5
+ import org.jetbrains.kotlinx.dataframe.api.ColsAtAnyDepthColumnsSelectionDsl.colsAtAnyDepth
6
+ import org.jetbrains.kotlinx.dataframe.api.FilterColumnsSelectionDsl.filter
5
7
import org.jetbrains.kotlinx.dataframe.api.add
6
8
import org.jetbrains.kotlinx.dataframe.api.after
7
9
import org.jetbrains.kotlinx.dataframe.api.chunked
@@ -590,7 +592,7 @@ class Access : TestBase() {
590
592
df.select { name.allCols() }
591
593
592
594
// 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() } }
594
596
595
597
// SampleEnd
596
598
}
@@ -626,7 +628,7 @@ class Access : TestBase() {
626
628
df.select { Person ::name.allCols() }
627
629
628
630
// 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() } }
630
632
// SampleEnd
631
633
}
632
634
@@ -660,7 +662,7 @@ class Access : TestBase() {
660
662
df.select { " name" .allCols() }
661
663
662
664
// 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() } }
664
666
// SampleEnd
665
667
}
666
668
@@ -719,13 +721,13 @@ class Access : TestBase() {
719
721
}
720
722
721
723
// traversal of columns at any depth from here excluding ColumnGroups
722
- df.select { colsAtAnyDepth { ! it.isColumnGroup() } }
724
+ df.select { colsAtAnyDepth().filter { ! it.isColumnGroup() } }
723
725
724
726
// traversal of columns at any depth from here including ColumnGroups
725
727
df.select { colsAtAnyDepth() }
726
728
727
729
// traversal of columns at any depth with condition
728
- df.select { colsAtAnyDepth { it.name().contains(" :" ) } }
730
+ df.select { colsAtAnyDepth().filter() { it.name().contains(" :" ) } }
729
731
730
732
// traversal of columns at any depth to find columns of given type
731
733
df.select { colsAtAnyDepth().colsOf<String >() }
@@ -743,18 +745,18 @@ class Access : TestBase() {
743
745
fun columnSelectorsModifySet () {
744
746
// SampleStart
745
747
// 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 ) }
748
750
749
751
// 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 ) }
752
754
753
755
// 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" ) } }
755
757
756
758
// exclude columns from column set
757
- df.select { colsAtAnyDepth { ! it.isColumnGroup() }.except { age } }
759
+ df.select { colsAtAnyDepth().filter { ! it.isColumnGroup() }.except { age } }
758
760
759
761
// keep only unique columns
760
762
df.select { (colsOf<Int >() and age).distinct() }
0 commit comments