regreplace

simonsobs · Dec 11, 2024 · 8977862 · 8977862
1 parent 73c19b4
commit 8977862
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pixell/utils.py b/pixell/utils.py
@@ -3467,6 +3467,11 @@ def replace(istr, ipat, repl):
 	if ostr == istr: raise KeyError("Pattern not found")
 	return ostr
 
+def regreplace(istr, ipat, repl, count=0, flags=0):
+	ostr, n = re.subn(ipat, repl, istr, count=count, flags=flags)
+	if n == 0: raise KeyError("Pattern not found")
+	return ostr
+
 # I used to do stuff like a[~np.isfinite(a)] = 0, but this should be
 # lower overhad and faster
 def remove_nan(a):