feat: Add `index_of()` function to `Series` and `Expr`#19894

Merged

ritchie46 merged 100 commits intopola-rs:mainfrom itamarst:5503-series-index_of

Jan 7, 2025

+619

Commits on Dec 16, 2024

Proof of concept end-to-end implementation of Series.index_of().

pythonspeed
authored and
coastalwhite
committed
Proof-of-concept NaN handling.

pythonspeed
authored and
coastalwhite
committed
Simplify.

pythonspeed
authored and
coastalwhite
committed
Minimal support for null values.

pythonspeed
authored and
coastalwhite
committed
Document the motivation.

pythonspeed
authored and
coastalwhite
committed
Integer casting.

pythonspeed
authored and
coastalwhite
committed
Implement lists.

pythonspeed
authored and
coastalwhite
committed
Support for arrays.

pythonspeed
authored and
coastalwhite
committed
Start refactoring to be in polars-ops and use expressions.

pythonspeed
authored and
coastalwhite
committed
Continued work on sketch of expr-based index_of().

pythonspeed
authored and
coastalwhite
committed
Forgot to check this on.

pythonspeed
authored and
coastalwhite
committed
Skip lists and arrays for now.

pythonspeed
authored and
coastalwhite
committed
Maybe fine as is?

pythonspeed
authored and
coastalwhite
committed
Only use the downcast macro once.

pythonspeed
authored and
coastalwhite
committed
Simplify by using AnyValue throughout.

pythonspeed
authored and
coastalwhite
committed
Not used

pythonspeed
authored and
coastalwhite
committed
Don't support multiple entries in this way

pythonspeed
authored and
coastalwhite
committed
Fast path for sorted values.

pythonspeed
authored and
coastalwhite
committed
Start of unit tests.

pythonspeed
authored and
coastalwhite
committed
search_sorted() can give (from our perspective) false positives.

pythonspeed
authored and
coastalwhite
committed
Null Series.

pythonspeed
authored and
coastalwhite
committed
Test the lazy API too.

pythonspeed
authored and
coastalwhite
committed
Basic integer tests.

pythonspeed
authored and
coastalwhite
committed
Fix bug where index_of() searched for the rounded value of floats.

pythonspeed
authored and
coastalwhite
committed
Another test

pythonspeed
authored and
coastalwhite
committed
Better phrasing

pythonspeed
authored and
coastalwhite
committed
Handle scalar columns.

pythonspeed
authored and
coastalwhite
committed
Hypothesis-based tests.

pythonspeed
authored and
coastalwhite
committed
Better explanation

pythonspeed
authored and
coastalwhite
committed
Fix some bugs and lints

pythonspeed
authored and
coastalwhite
committed
Add a feature.

pythonspeed
authored and
coastalwhite
committed
Improve docs and type annotations.

pythonspeed
authored and
coastalwhite
committed
Lint fix.

pythonspeed
authored and
coastalwhite
committed
Catch Python stdlib floats too.

pythonspeed
authored and
coastalwhite
committed
Start moving towards code sharing.

pythonspeed
authored and
coastalwhite
committed
Start removing usage of ChunkedArray.iter().

pythonspeed
authored and
coastalwhite
committed
Remove more usage of iter().

pythonspeed
authored and
coastalwhite
committed
Pacify clippy.

pythonspeed
authored and
coastalwhite
committed
Not used.

pythonspeed
authored and
coastalwhite
committed
Improve test coverage.

pythonspeed
authored and
coastalwhite
committed
Fix lint.

pythonspeed
authored and
coastalwhite
committed
Pacify clippy

pythonspeed
authored and
coastalwhite
committed
Only load if feature enabled.

pythonspeed
authored and
coastalwhite
committed
Reformat.

pythonspeed
authored and
coastalwhite
committed
Documentation improvements.

pythonspeed
authored and
coastalwhite
committed
Fix location.

pythonspeed
authored and
coastalwhite
committed
Get rid of trait.

pythonspeed
authored and
coastalwhite
committed
Simplify by using TotalEq.

pythonspeed
authored and
coastalwhite
committed
Do supertype casting at a higher level.

pythonspeed
authored and
coastalwhite
committed
Sketch of row-encoding based search for non-numeric data types.

pythonspeed
authored and
coastalwhite
committed
Fix searching for nulls (and speed it up!) by moving it earlier in the processing.

pythonspeed
authored and
coastalwhite
committed
Basic tests for other dtypes

pythonspeed
authored and
coastalwhite
committed
Improve comments.

pythonspeed
authored and
coastalwhite
committed
Fix decimals.

pythonspeed
authored and
coastalwhite
committed
Fix lists and arrays.

pythonspeed
authored and
coastalwhite
committed
Lint

pythonspeed
authored and
coastalwhite
committed
Improve test coverage.

pythonspeed
authored and
coastalwhite
committed
Handle edge case where we pass in an array literal, so the series is kept as an array instead of being converted to a list.

pythonspeed
authored and
coastalwhite
committed
Gate on feature

pythonspeed
authored and
coastalwhite
committed
Skip enum and categorical for now

pythonspeed
authored and
coastalwhite
committed
Fix some lints

pythonspeed
authored and
coastalwhite
committed
Fix lint

pythonspeed
authored and
coastalwhite
committed
Only cast value in edge case that requires it.

pythonspeed
authored and
coastalwhite
committed
Remove unnecessary trait

pythonspeed
authored and
coastalwhite
committed
Arrays and Lists are now better supported in row encoding

pythonspeed
authored and
coastalwhite
committed
Test a couple more edge cases

pythonspeed
authored and
coastalwhite
committed
Fix formatting.

pythonspeed
authored and
coastalwhite
committed
fix small rebase error
coastalwhite
committed
fix: Properly handle no item find in group-by
coastalwhite
committed
fix: Error on multiple values
coastalwhite
committed
fix: Properly error on type mismatch
coastalwhite
committed
Less code duplication.
pythonspeed
committed
Better tests for Enum and Categorical
pythonspeed
committed
Point at the correct issue
pythonspeed
committed
Restore missing coverage
pythonspeed
committed
Null can be cast to anything.
pythonspeed
committed
Merge remote-tracking branch 'origin/main' into 5503-series-index_of
pythonspeed
committed
Update to latest API.
pythonspeed
committed
Give good error messages, which can be removed when corresponding bugs are fixed.
pythonspeed
committed
Error out instead of giving the wrong result
pythonspeed
committed
Format
pythonspeed
committed