public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-05 16:49:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Don't have Matrix try to convert lists of Mobject into numpy arrays

Browse source 
https://github.com/3b1b/manim/issues/1878
This commit is contained in:
Grant Sanderson 2022-11-03 16:31:19 -07:00
parent 8e6265d35e
commit f878537814
1 changed files with 2 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
manimlib/mobject/matrix.py
View file

@ -94,15 +94,12 @@ class Matrix(VMobject):
or mobjects or mobjects
""" """
VMobject.__init__(self, **kwargs) VMobject.__init__(self, **kwargs)
matrix = self.matrix = np.array(matrix, ndmin=2) mob_matrix = self.mob_matrix = self.matrix_to_mob_matrix(matrix)
mob_matrix = self.matrix_to_mob_matrix(matrix)
self.organize_mob_matrix(mob_matrix) self.organize_mob_matrix(mob_matrix)
# self.elements = VGroup(*mob_matrix.flatten())
self.elements = VGroup(*it.chain(*mob_matrix)) self.elements = VGroup(*it.chain(*mob_matrix))
self.add(self.elements) self.add(self.elements)
self.add_brackets() self.add_brackets()
self.center() self.center()
self.mob_matrix = mob_matrix
if self.add_background_rectangles_to_entries: if self.add_background_rectangles_to_entries:
for mob in self.elements: for mob in self.elements:
mob.add_background_rectangle() mob.add_background_rectangle()
@ -129,7 +126,7 @@ class Matrix(VMobject):
return self return self
def add_brackets(self): def add_brackets(self):
height = self.matrix.shape[0] height = len(self.mob_matrix)
bracket_pair = Tex("".join([ bracket_pair = Tex("".join([
"\\left[", "\\left[",
"\\begin{array}{c}", "\\begin{array}{c}",