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

Replacing apply_along_axis

Browse Source
This commit is contained in:
Grant Sanderson
2020-02-19 23:43:33 -08:00
parent 0176bda584
commit c794039b9d
4 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
manimlib/mobject/mobject.py
View File

@ -298,7 +298,7 @@ class Mobject(Container):
if len(kwargs) == 0:
kwargs["about_point"] = ORIGIN
self.apply_points_function_about_point(
lambda points: np.apply_along_axis(function, 1, points),
lambda points: np.array([function(p) for p in points]),
**kwargs
)
return self
@ -775,7 +775,7 @@ class Mobject(Container):
return self.get_bounding_box_point(np.zeros(self.dim))
def get_center_of_mass(self):
return np.apply_along_axis(np.mean, 0, self.get_all_points())
return self.get_all_points().mean(0)
def get_boundary_point(self, direction):
all_points = self.get_points_defining_boundary()