extern/nushell
1
0
Fork 1
mirror of https://github.com/nushell/nushell.git synced 2025-06-11 20:46:48 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add regex documentation/examples to polars col (#15898)

Browse Source 
# Description
Include regular expression example and help documentation to `polars
col`

Co-authored-by: Jack Wright <jack.wright@nike.com>
This commit is contained in:
Jack Wright 2025-06-05 13:53:22 -07:00 committed by GitHub
parent 4a9e2ac37b
commit 21d949207f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 17 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
crates/nu_plugin_polars/src/dataframe/command/data/col.rs
View File

@ -8,7 +8,7 @@ use nu_protocol::{
Category, Example, LabeledError, PipelineData, ShellError, Signature, Span, SyntaxShape, Type,
Value, record,
};
use polars::prelude::DataType;
use polars::{df, prelude::DataType};
#[derive(Clone)]
pub struct ExprCol;
@ -29,7 +29,7 @@ impl PluginCommand for ExprCol {
.required(
"column name",
SyntaxShape::String,
"Name of column to be used. '*' can be used for all columns.",
"Name of column to be used. '*' can be used for all columns. Accepts regular expression input; regular expressions should start with ^ and end with $.",
)
.rest(
"more columns",
@ -107,6 +107,21 @@ impl PluginCommand for ExprCol {
.into_value(Span::test_data()),
),
},
Example {
description: "Select columns using a regular expression",
example: "[[ham hamburger foo bar]; [1 11 2 a] [2 22 1 b]] | polars into-df | polars select (polars col '^ham.*$') | polars collect",
result: Some(
NuDataFrame::new(
false,
df!(
"ham" => [1, 2],
"hamburger" => [11, 22],
)
.expect("should not fail to create dataframe"),
)
.into_value(Span::test_data()),
),
},
]
}