https://github.com/ion-elgreco/polars-distance
Polars plugin for pairwise distance functions
https://github.com/ion-elgreco/polars-distance
Last synced: 2 months ago
JSON representation
Polars plugin for pairwise distance functions
- Host: GitHub
- URL: https://github.com/ion-elgreco/polars-distance
- Owner: ion-elgreco
- License: mit
- Created: 2023-11-04T23:27:41.000Z (over 1 year ago)
- Default Branch: main
- Last Pushed: 2025-03-07T15:40:14.000Z (3 months ago)
- Last Synced: 2025-03-28T15:04:24.498Z (2 months ago)
- Language: Rust
- Homepage: https://ion-elgreco.github.io/polars-distance/
- Size: 733 KB
- Stars: 65
- Watchers: 1
- Forks: 10
- Open Issues: 3
-
Metadata Files:
- Readme: README.md
- Contributing: CONTRIBUTING.md
- License: LICENSE
Awesome Lists containing this project
- awesome-polars - polars-distance - Polars plugin for text similarity/pairwise distance functions by [@ion-elgreco](https://github.com/ion-elgreco). (Libraries/Packages/Scripts / Polars plugins)
README
This plugin provides distance metrics on list, arrays and string datatypes in polars.
The docs can be found here: https://ion-elgreco.github.io/polars-distance/
## Examples
```python
import polars as pl
import polars_distance as pld
df = pl.DataFrame({
"foo":"hello",
"bar":"hella world"
})
df.select(
pld.col("foo").dist_str.hamming('bar').alias('dist')
)
┌──────┐
│ dist │
│ --- │
│ u32 │
╞══════╡
│ 7 │
└──────┘
df.select(
pld.col('foo').dist_str.levenshtein('bar').alias('dist')
)
┌──────┐
│ dist │
│ --- │
│ u32 │
╞══════╡
│ 6 │
└──────┘
df = pl.DataFrame(
{
"arr": [[1, 2, 10]],
"arr2": [[2, 5, 9]],
},
schema={
"arr": pl.Array(inner=pl.Float64, width=3),
"arr2": pl.Array(inner=pl.Float64, width=3),
},
)
df.select(pld.col('arr').dist_arr.euclidean('arr2').alias('dist'))
shape: (1, 1)
┌──────────┐
│ dist │
│ --- │
│ f64 │
╞══════════╡
│ 3.316625 │
└──────────┘
```