Fixed get_merged_array bug

This commit is contained in:
Grant Sanderson 2018-01-15 15:06:01 -08:00
parent ec1afdbb83
commit 5d9eb8d5a0
2 changed files with 5 additions and 3 deletions

View file

@ -496,8 +496,11 @@ class Mobject(object):
return 0 return 0
def get_merged_array(self, array_attr): def get_merged_array(self, array_attr):
result = np.zeros((0, self.dim)) result = None
for mob in self.family_members_with_points(): for mob in self.family_members_with_points():
if result is None:
result = getattr(mob, array_attr)
else:
result = np.append(result, getattr(mob, array_attr), 0) result = np.append(result, getattr(mob, array_attr), 0)
return result return result

View file

@ -47,7 +47,6 @@ class PMobject(Mobject):
]) ])
return self return self
def match_colors(self, mobject): def match_colors(self, mobject):
Mobject.align_data(self, mobject) Mobject.align_data(self, mobject)
self.rgbas = np.array(mobject.rgbas) self.rgbas = np.array(mobject.rgbas)