rmin and rmax

shanedrabing · Jul 21, 2021 · 38ed832 · 38ed832
1 parent 1e69352
commit 38ed832
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 0 deletions.
diff --git a/pyrat/base.py b/pyrat/base.py
@@ -38,6 +38,8 @@
     "unique",
     "vector",
     "which",
+    "rmin",
+    "rmax",
 ]
 
 
@@ -132,6 +134,28 @@ def sqrt(x):
     return x.apply(catch(math.sqrt, TypeError, NA))
 
 
+def rmin(*x, na_rm=False):
+    x = c(x)
+    if not x:
+        return Inf
+    if na_rm:
+        x = x[~is_na(x)]
+    elif any(is_na(x)):
+        return NA
+    return min(x)
+
+
+def rmax(*x, na_rm=False):
+    x = c(x)
+    if not x:
+        return -Inf
+    if na_rm:
+        x = x[~is_na(x)]
+    elif any(is_na(x)):
+        return NA
+    return max(x)
+
+
 def mean(x, na_rm=False):
     if not x:
         return NA

diff --git a/tests.py b/tests.py
@@ -313,3 +313,14 @@ def none(x):
     assert_identical(mean(c(1, 2)), 1.5)
     assert_identical(mean(c(1, 2, NA)), NA)
     assert_identical(mean(c(1, 2, NA), na_rm=True), 1.5)
+
+    # rmin
+    assert_eq(rmin(), Inf)
+    assert_eq(rmin(1, c(NA, 2)), NA)
+    assert_eq(rmin(1, NA, 2, na_rm=True), 1)
+
+    # rmax
+    assert_eq(rmax(), -Inf)
+    assert_eq(rmax(1, c(NA, 2)), NA)
+    assert_eq(rmax(1, NA, 2, na_rm=True), 2)
+