github/manim
1
1
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2026-04-26 03:00:23 -04:00
Code Issues Packages Projects Releases Wiki Activity

Added complex_func_to_R3_func

Browse Source
This commit is contained in:
Grant Sanderson
2018-05-30 12:02:25 -07:00
parent 49fe36b048
commit 03cc70b713
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
utils/space_ops.py
View File

@@ -110,6 +110,10 @@ def R3_to_complex(point):
return complex(*point[:2])
def complex_func_to_R3_func(complex_func):
return lambda p: complex_to_R3(complex_func(R3_to_complex(p)))
def center_of_mass(points):
points = [np.array(point).astype("float") for point in points]
return sum(points) / len(points)