Change some defaults and add Matrix.get_rows method

This commit is contained in:
Grant Sanderson 2021-03-27 11:56:58 -07:00
parent d3e61b962b
commit 7f47815230

View file

@ -57,8 +57,8 @@ class Matrix(VMobject):
CONFIG = { CONFIG = {
"v_buff": 0.8, "v_buff": 0.8,
"h_buff": 1.3, "h_buff": 1.3,
"bracket_h_buff": MED_SMALL_BUFF, "bracket_h_buff": 0.2,
"bracket_v_buff": MED_SMALL_BUFF, "bracket_v_buff": 0.25,
"add_background_rectangles_to_entries": False, "add_background_rectangles_to_entries": False,
"include_background_rectangle": False, "include_background_rectangle": False,
"element_to_mobject": Tex, "element_to_mobject": Tex,
@ -132,6 +132,12 @@ class Matrix(VMobject):
for i in range(len(self.mob_matrix[0])) for i in range(len(self.mob_matrix[0]))
]) ])
def get_rows(self):
return VGroup(*[
VGroup(*row)
for row in self.mob_matrix
])
def set_column_colors(self, *colors): def set_column_colors(self, *colors):
columns = self.get_columns() columns = self.get_columns()
for color, column in zip(colors, columns): for color, column in zip(colors, columns):