From 77a2316bcff06acf7c33b6da6ac7a112fde44c40 Mon Sep 17 00:00:00 2001 From: wangfeng Date: Wed, 1 Feb 2023 21:52:50 +0800 Subject: [PATCH] =?UTF-8?q?#I6C994:=20=E6=94=AF=E6=8C=81dataframe=E9=80=89?= =?UTF-8?q?=E4=B8=80=E6=AE=B5=E5=AD=90=E9=9B=86,=20=E7=9B=AE=E5=89=8D?= =?UTF-8?q?=E6=9A=82=E4=B8=8D=E6=94=AF=E6=8C=81=E7=B4=A2=E5=BC=95,=20?= =?UTF-8?q?=E5=8F=AA=E8=83=BD=E6=8C=89=E7=85=A7=E8=A1=8C=E5=8F=B7=E5=8C=BA?= =?UTF-8?q?=E9=97=B4=E6=9D=A5=E9=80=89=E6=8B=A9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- dataframe.go | 22 ---------------------- dataframe_subset.go | 23 +++++++++++++++++++++++ 2 files changed, 23 insertions(+), 22 deletions(-) create mode 100644 dataframe_subset.go diff --git a/dataframe.go b/dataframe.go index ff61dbf..2f7d827 100644 --- a/dataframe.go +++ b/dataframe.go @@ -228,28 +228,6 @@ func (df DataFrame) print( return str } -// Subset returns a subset of the rows of the original DataFrame based on the -// Series subsetting indexes. -func (df DataFrame) Subset(start, end int) DataFrame { - if df.Err != nil { - return df - } - columns := make([]Series, df.ncols) - for i, column := range df.columns { - s := column.Subset(start, end) - columns[i] = s - } - nrows, ncols, err := checkColumnsDimensions(columns...) - if err != nil { - return DataFrame{Err: err} - } - return DataFrame{ - columns: columns, - ncols: ncols, - nrows: nrows, - } -} - func checkColumnsDimensions(se ...Series) (nrows, ncols int, err error) { ncols = len(se) nrows = -1 diff --git a/dataframe_subset.go b/dataframe_subset.go new file mode 100644 index 0000000..ea0b0d9 --- /dev/null +++ b/dataframe_subset.go @@ -0,0 +1,23 @@ +package pandas + +// Subset returns a subset of the rows of the original DataFrame based on the +// Series subsetting indexes. +func (df DataFrame) Subset(start, end int) DataFrame { + if df.Err != nil { + return df + } + columns := make([]Series, df.ncols) + for i, column := range df.columns { + s := column.Subset(start, end) + columns[i] = s + } + nrows, ncols, err := checkColumnsDimensions(columns...) + if err != nil { + return DataFrame{Err: err} + } + return DataFrame{ + columns: columns, + ncols: ncols, + nrows: nrows, + } +} -- Gitee