opensource/manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-07-28 12:32:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add arr_clip (faster alternative to np.clip)

Browse Source
This commit is contained in:
Grant Sanderson
2023-01-08 21:27:56 -05:00
parent d5fdc75164
commit 9e7cd1399d
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
manimlib/utils/simple_functions.py
View File

@ -51,6 +51,12 @@ def clip(a: float, min_a: float, max_a: float) -> float:
return a return a
def arr_clip(arr: np.ndarray, min_a: float, max_a: float) -> np.ndarray:
arr[arr < min_a] = min_a
arr[arr > max_a] = max_a
return arr
def fdiv(a: Scalable, b: Scalable, zero_over_zero_value: Scalable | None = None) -> Scalable: def fdiv(a: Scalable, b: Scalable, zero_over_zero_value: Scalable | None = None) -> Scalable:
if zero_over_zero_value is not None: if zero_over_zero_value is not None:
out = np.full_like(a, zero_over_zero_value) out = np.full_like(a, zero_over_zero_value)