From d443ce8f5ce268b6039b27a79544b0ef649f284f Mon Sep 17 00:00:00 2001 From: Patrick Hoefler Date: Tue, 16 Aug 2022 11:21:41 +0200 Subject: [PATCH 1/5] ASV: Add asv for clip with date range --- asv_bench/benchmarks/series_methods.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/asv_bench/benchmarks/series_methods.py b/asv_bench/benchmarks/series_methods.py index 1d3bd4a357d24..7821cb4d975d8 100644 --- a/asv_bench/benchmarks/series_methods.py +++ b/asv_bench/benchmarks/series_methods.py @@ -139,10 +139,16 @@ class Clip: def setup(self, n): self.s = Series(np.random.randn(n)) + dr = date_range("20220101", periods=500_000, freq="s", tz="UTC") + self.clipper_dt = dr[0:1000].repeat(100) + self.ser_dt = Series(dr) def time_clip(self, n): self.s.clip(0, 1) + def time_clip_dt(self, n): + self.ser_dt.clip(upper=self.clipper_dt) + class ValueCounts: From fa650b994dbe2c4b555c3d431fac0c4df8dc9438 Mon Sep 17 00:00:00 2001 From: Patrick Hoefler Date: Tue, 16 Aug 2022 11:21:41 +0200 Subject: [PATCH 2/5] ASV: Add asv for clip with date range ASV: Add asv for clip with date range --- asv_bench/benchmarks/series_methods.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/asv_bench/benchmarks/series_methods.py b/asv_bench/benchmarks/series_methods.py index 7821cb4d975d8..af6dfe7c2e321 100644 --- a/asv_bench/benchmarks/series_methods.py +++ b/asv_bench/benchmarks/series_methods.py @@ -139,7 +139,7 @@ class Clip: def setup(self, n): self.s = Series(np.random.randn(n)) - dr = date_range("20220101", periods=500_000, freq="s", tz="UTC") + dr = date_range("20220101", periods=n, freq="s", tz="UTC") self.clipper_dt = dr[0:1000].repeat(100) self.ser_dt = Series(dr) From 928352bccbb6430b1eb2d45e107347e0b1896c30 Mon Sep 17 00:00:00 2001 From: Patrick Hoefler Date: Tue, 16 Aug 2022 12:52:16 +0200 Subject: [PATCH 3/5] Fix asv --- asv_bench/benchmarks/series_methods.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/asv_bench/benchmarks/series_methods.py b/asv_bench/benchmarks/series_methods.py index af6dfe7c2e321..e1f6f5e4401c3 100644 --- a/asv_bench/benchmarks/series_methods.py +++ b/asv_bench/benchmarks/series_methods.py @@ -140,7 +140,7 @@ class Clip: def setup(self, n): self.s = Series(np.random.randn(n)) dr = date_range("20220101", periods=n, freq="s", tz="UTC") - self.clipper_dt = dr[0:1000].repeat(100) + self.clipper_dt = dr[0 : min(n, 100)].repeat(10) self.ser_dt = Series(dr) def time_clip(self, n): @@ -150,6 +150,16 @@ def time_clip_dt(self, n): self.ser_dt.clip(upper=self.clipper_dt) +class ClipDt: + def setup(self, n): + dr = date_range("20220101", periods=100_000, freq="s", tz="UTC") + self.clipper_dt = dr[0:1000].repeat(100) + self.s = Series(dr) + + def time_clip(self, n): + self.s.clip(upper=self.clipper_dt) + + class ValueCounts: params = [[10**3, 10**4, 10**5], ["int", "uint", "float", "object"]] From e2d26c50c142961a94a12e44529bba2115a40e69 Mon Sep 17 00:00:00 2001 From: Patrick Hoefler Date: Tue, 16 Aug 2022 15:55:23 +0200 Subject: [PATCH 4/5] Remove --- asv_bench/benchmarks/series_methods.py | 6 ------ 1 file changed, 6 deletions(-) diff --git a/asv_bench/benchmarks/series_methods.py b/asv_bench/benchmarks/series_methods.py index e1f6f5e4401c3..45791f25db1bf 100644 --- a/asv_bench/benchmarks/series_methods.py +++ b/asv_bench/benchmarks/series_methods.py @@ -139,16 +139,10 @@ class Clip: def setup(self, n): self.s = Series(np.random.randn(n)) - dr = date_range("20220101", periods=n, freq="s", tz="UTC") - self.clipper_dt = dr[0 : min(n, 100)].repeat(10) - self.ser_dt = Series(dr) def time_clip(self, n): self.s.clip(0, 1) - def time_clip_dt(self, n): - self.ser_dt.clip(upper=self.clipper_dt) - class ClipDt: def setup(self, n): From 7752af7d998b4adf87c00d0b847942c99107db64 Mon Sep 17 00:00:00 2001 From: Patrick Hoefler Date: Tue, 16 Aug 2022 17:34:08 +0200 Subject: [PATCH 5/5] Remove type --- asv_bench/benchmarks/series_methods.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/asv_bench/benchmarks/series_methods.py b/asv_bench/benchmarks/series_methods.py index 45791f25db1bf..09c318af76159 100644 --- a/asv_bench/benchmarks/series_methods.py +++ b/asv_bench/benchmarks/series_methods.py @@ -145,12 +145,12 @@ def time_clip(self, n): class ClipDt: - def setup(self, n): + def setup(self): dr = date_range("20220101", periods=100_000, freq="s", tz="UTC") - self.clipper_dt = dr[0:1000].repeat(100) + self.clipper_dt = dr[0:1_000].repeat(100) self.s = Series(dr) - def time_clip(self, n): + def time_clip(self): self.s.clip(upper=self.clipper_dt)