diff --git a/mobject/matrix.py b/mobject/matrix.py index e65674ad..2c1fe1dd 100644 --- a/mobject/matrix.py +++ b/mobject/matrix.py @@ -86,8 +86,7 @@ class Matrix(VMobject): for mob in self.elements: mob.add_background_rectangle() if self.include_background_rectangle: - self.background_rectangle = BackgroundRectangle(self) - self.add_to_back(self.background_rectangle) + self.add_background_rectangle() def matrix_to_mob_matrix(self, matrix): return np.vectorize( @@ -117,6 +116,10 @@ class Matrix(VMobject): self.brackets = VGroup(l_bracket, r_bracket) return self + def add_background_rectangle(self): + self.background_rectangle = BackgroundRectangle(self) + self.add_to_back(self.background_rectangle) + def set_color_columns(self, *colors): for i, color in enumerate(colors): VGroup(*self.mob_matrix[:, i]).set_color(color)