mirror of
https://github.com/nushell/nushell.git
synced 2025-05-31 23:26:11 +02:00
966cebec34
# Description
This PR adds the `polars list-contains` command. It works like this:
```
~/Projects/nushell/nushell> let df = [[a]; [[a,b,c]] [[b,c,d]] [[c,d,f]]] | polars into-df -s {a: list<str>};
~/Projects/nushell/nushell> $df | polars with-column [(polars col a | polars list-contains (polars lit a) | polars as b)] | polars collect
╭───┬───────────┬───────╮
│ # │ a │ b │
├───┼───────────┼───────┤
│ 0 │ ╭───┬───╮ │ true │
│ │ │ 0 │ a │ │ │
│ │ │ 1 │ b │ │ │
│ │ │ 2 │ c │ │ │
│ │ ╰───┴───╯ │ │
│ 1 │ ╭───┬───╮ │ false │
│ │ │ 0 │ b │ │ │
│ │ │ 1 │ c │ │ │
│ │ │ 2 │ d │ │ │
│ │ ╰───┴───╯ │ │
│ 2 │ ╭───┬───╮ │ false │
│ │ │ 0 │ c │ │ │
│ │ │ 1 │ d │ │ │
│ │ │ 2 │ f │ │ │
│ │ ╰───┴───╯ │ │
╰───┴───────────┴───────╯
```
or
```
~/Projects/nushell/nushell> let df = [[a, b]; [[a,b,c], a] [[b,c,d], f] [[c,d,f], f]] | polars into-df -s {a: list<str>, b: str}
~/Projects/nushell/nushell> $df | polars with-column [(polars col a | polars list-contains b | polars as c)] | polars collect
╭───┬───────────┬───┬───────╮
│ # │ a │ b │ c │
├───┼───────────┼───┼───────┤
│ 0 │ ╭───┬───╮ │ a │ true │
│ │ │ 0 │ a │ │ │ │
│ │ │ 1 │ b │ │ │ │
│ │ │ 2 │ c │ │ │ │
│ │ ╰───┴───╯ │ │ │
│ 1 │ ╭───┬───╮ │ f │ false │
│ │ │ 0 │ b │ │ │ │
│ │ │ 1 │ c │ │ │ │
│ │ │ 2 │ d │ │ │ │
│ │ ╰───┴───╯ │ │ │
│ 2 │ ╭───┬───╮ │ f │ true │
│ │ │ 0 │ c │ │ │ │
│ │ │ 1 │ d │ │ │ │
│ │ │ 2 │ f │ │ │ │
│ │ ╰───┴───╯ │ │ │
╰───┴───────────┴───┴───────╯
```
or
```
~/Projects/nushell/nushell> let df = [[a, b]; [[1,2,3], 4] [[2,4,1], 2] [[2,1,6], 3]] | polars into-df -s {a: list<i64>, b: i64}
~/Projects/nushell/nushell> $df | polars with-column [(polars col a | polars list-contains ((polars col b) * 2) | polars as c)] | polars collect
╭───┬───────────┬───┬───────╮
│ # │ a │ b │ c │
├───┼───────────┼───┼───────┤
│ 0 │ ╭───┬───╮ │ 4 │ false │
│ │ │ 0 │ 1 │ │ │ │
│ │ │ 1 │ 2 │ │ │ │
│ │ │ 2 │ 3 │ │ │ │
│ │ ╰───┴───╯ │ │ │
│ 1 │ ╭───┬───╮ │ 2 │ true │
│ │ │ 0 │ 2 │ │ │ │
│ │ │ 1 │ 4 │ │ │ │
│ │ │ 2 │ 1 │ │ │ │
│ │ ╰───┴───╯ │ │ │
│ 2 │ ╭───┬───╮ │ 3 │ true │
│ │ │ 0 │ 2 │ │ │ │
│ │ │ 1 │ 1 │ │ │ │
│ │ │ 2 │ 6 │ │ │ │
│ │ ╰───┴───╯ │ │ │
╰───┴───────────┴───┴───────╯
```
Let me know what you think. I'm a bit surprised that a list by default
seems to get converted to "object" when doing `into-df` which is why I
added the extra `-s` flag every time to explicitly force it into a list.