opensource/manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-07-30 13:34:19 +08: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]) 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): def center_of_mass(points):
points = [np.array(point).astype("float") for point in points] points = [np.array(point).astype("float") for point in points]
return sum(points) / len(points) return sum(points) / len(points)