public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-09-01 00:48:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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
Show stats 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
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:
if zero_over_zero_value is not None:
out = np.full_like(a, zero_over_zero_value)