From 03cc70b71338de8beea9fd11fecd80a4fa64fe18 Mon Sep 17 00:00:00 2001 From: Grant Sanderson Date: Wed, 30 May 2018 12:02:25 -0700 Subject: [PATCH] Added complex_func_to_R3_func --- utils/space_ops.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/utils/space_ops.py b/utils/space_ops.py index 792d9539..a55485ed 100644 --- a/utils/space_ops.py +++ b/utils/space_ops.py @@ -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)