矩阵与表格¶

def __init__(
    self,
    matrix: Entries,
    v_buff: float = 0.8,
    h_buff: float = 1.3,
    bracket_h_buff: float = MED_SMALL_BUFF,
    bracket_v_buff: float = MED_SMALL_BUFF,
    add_background_rectangles_to_entries: bool = False,
    include_background_rectangle: bool = False,
    element_to_mobject: ElementFactory | None = None,
    element_to_mobject_config: Mapping[str, object] | None = None,
    element_alignment_corner: Vector3DLike = DR,
    left_bracket: str = "[",
    right_bracket: str = "]",
    stretch_brackets: bool = True,
    bracket_config: MathTexOptions | None = None,
    **kwargs: Unpack[Style],
):
    if bracket_config is None:
        bracket_config = {}
    super().__init__(**kwargs)
    self.cells = self._make_cells(
        matrix, element_to_mobject, element_to_mobject_config
    )
    cells = self.cells
    for i, row in enumerate(cells):
        for j, _ in enumerate(row):
            mob = cells[i][j]
            mob.move_to(
                i * v_buff * DOWN + j * h_buff * RIGHT,
                element_alignment_corner,
            )
    self.elements = VGroup(*it.chain(*self.cells))
    """条目，逐行排列：以组的形式组织，并被添加到矩阵中。"""
    self.add(self.elements)
    bracket_config = {**bracket_config}
    BRACKET_HEIGHT = 0.5977
    n = int(self.height / BRACKET_HEIGHT) + 1
    empty_tex_array = "".join(
        ["\\begin{array}{c}", *n * ["\\quad \\\\"], "\\end{array}"]
    )
    tex_left = "".join(["\\left" + left_bracket, empty_tex_array, "\\right."])
    tex_right = "".join(["\\left.", empty_tex_array, "\\right" + right_bracket])
    l_bracket = MathTex(tex_left, **bracket_config)
    r_bracket = MathTex(tex_right, **bracket_config)
    bracket_pair = VGroup(l_bracket, r_bracket)
    if stretch_brackets:
        bracket_pair.stretch_to_fit_height(self.height + 2 * bracket_v_buff)
    l_bracket.next_to(self, LEFT, bracket_h_buff)
    r_bracket.next_to(self, RIGHT, bracket_h_buff)
    self.brackets = bracket_pair
    self.add(l_bracket, r_bracket)
    self.center()
    if add_background_rectangles_to_entries:
        for mob in self.elements:
            mob.add_background_rectangle()
    if include_background_rectangle:
        self.add_background_rectangle()

def __init__(
    self,
    matrix: Entries,
    v_buff: float = 0.8,
    h_buff: float = 1.3,
    bracket_h_buff: float = MED_SMALL_BUFF,
    bracket_v_buff: float = MED_SMALL_BUFF,
    add_background_rectangles_to_entries: bool = False,
    include_background_rectangle: bool = False,
    element_to_mobject: ElementFactory | None = None,
    element_to_mobject_config: Mapping[str, object] | None = None,
    element_alignment_corner: Vector3DLike = DR,
    left_bracket: str = "[",
    right_bracket: str = "]",
    stretch_brackets: bool = True,
    bracket_config: MathTexOptions | None = None,
    **kwargs: Unpack[Style],
):
    if bracket_config is None:
        bracket_config = {}
    super().__init__(**kwargs)
    self.cells = self._make_cells(
        matrix, element_to_mobject, element_to_mobject_config
    )
    cells = self.cells
    for i, row in enumerate(cells):
        for j, _ in enumerate(row):
            mob = cells[i][j]
            mob.move_to(
                i * v_buff * DOWN + j * h_buff * RIGHT,
                element_alignment_corner,
            )
    self.elements = VGroup(*it.chain(*self.cells))
    """条目，逐行排列：以组的形式组织，并被添加到矩阵中。"""
    self.add(self.elements)
    bracket_config = {**bracket_config}
    BRACKET_HEIGHT = 0.5977
    n = int(self.height / BRACKET_HEIGHT) + 1
    empty_tex_array = "".join(
        ["\\begin{array}{c}", *n * ["\\quad \\\\"], "\\end{array}"]
    )
    tex_left = "".join(["\\left" + left_bracket, empty_tex_array, "\\right."])
    tex_right = "".join(["\\left.", empty_tex_array, "\\right" + right_bracket])
    l_bracket = MathTex(tex_left, **bracket_config)
    r_bracket = MathTex(tex_right, **bracket_config)
    bracket_pair = VGroup(l_bracket, r_bracket)
    if stretch_brackets:
        bracket_pair.stretch_to_fit_height(self.height + 2 * bracket_v_buff)
    l_bracket.next_to(self, LEFT, bracket_h_buff)
    r_bracket.next_to(self, RIGHT, bracket_h_buff)
    self.brackets = bracket_pair
    self.add(l_bracket, r_bracket)
    self.center()
    if add_background_rectangles_to_entries:
        for mob in self.elements:
            mob.add_background_rectangle()
    if include_background_rectangle:
        self.add_background_rectangle()

def __init__(
    self,
    matrix: Entries,
    v_buff: float = 0.8,
    h_buff: float = 1.3,
    bracket_h_buff: float = MED_SMALL_BUFF,
    bracket_v_buff: float = MED_SMALL_BUFF,
    add_background_rectangles_to_entries: bool = False,
    include_background_rectangle: bool = False,
    element_to_mobject: ElementFactory | None = None,
    element_to_mobject_config: Mapping[str, object] | None = None,
    element_alignment_corner: Vector3DLike = DR,
    left_bracket: str = "[",
    right_bracket: str = "]",
    stretch_brackets: bool = True,
    bracket_config: MathTexOptions | None = None,
    **kwargs: Unpack[Style],
):
    if bracket_config is None:
        bracket_config = {}
    super().__init__(**kwargs)
    self.cells = self._make_cells(
        matrix, element_to_mobject, element_to_mobject_config
    )
    cells = self.cells
    for i, row in enumerate(cells):
        for j, _ in enumerate(row):
            mob = cells[i][j]
            mob.move_to(
                i * v_buff * DOWN + j * h_buff * RIGHT,
                element_alignment_corner,
            )
    self.elements = VGroup(*it.chain(*self.cells))
    """条目，逐行排列：以组的形式组织，并被添加到矩阵中。"""
    self.add(self.elements)
    bracket_config = {**bracket_config}
    BRACKET_HEIGHT = 0.5977
    n = int(self.height / BRACKET_HEIGHT) + 1
    empty_tex_array = "".join(
        ["\\begin{array}{c}", *n * ["\\quad \\\\"], "\\end{array}"]
    )
    tex_left = "".join(["\\left" + left_bracket, empty_tex_array, "\\right."])
    tex_right = "".join(["\\left.", empty_tex_array, "\\right" + right_bracket])
    l_bracket = MathTex(tex_left, **bracket_config)
    r_bracket = MathTex(tex_right, **bracket_config)
    bracket_pair = VGroup(l_bracket, r_bracket)
    if stretch_brackets:
        bracket_pair.stretch_to_fit_height(self.height + 2 * bracket_v_buff)
    l_bracket.next_to(self, LEFT, bracket_h_buff)
    r_bracket.next_to(self, RIGHT, bracket_h_buff)
    self.brackets = bracket_pair
    self.add(l_bracket, r_bracket)
    self.center()
    if add_background_rectangles_to_entries:
        for mob in self.elements:
            mob.add_background_rectangle()
    if include_background_rectangle:
        self.add_background_rectangle()

def __init__(
    self,
    matrix: Entries,
    v_buff: float = 0.8,
    h_buff: float = 1.3,
    bracket_h_buff: float = MED_SMALL_BUFF,
    bracket_v_buff: float = MED_SMALL_BUFF,
    add_background_rectangles_to_entries: bool = False,
    include_background_rectangle: bool = False,
    element_to_mobject: ElementFactory | None = None,
    element_to_mobject_config: Mapping[str, object] | None = None,
    element_alignment_corner: Vector3DLike = DR,
    left_bracket: str = "[",
    right_bracket: str = "]",
    stretch_brackets: bool = True,
    bracket_config: MathTexOptions | None = None,
    **kwargs: Unpack[Style],
):
    if bracket_config is None:
        bracket_config = {}
    super().__init__(**kwargs)
    self.cells = self._make_cells(
        matrix, element_to_mobject, element_to_mobject_config
    )
    cells = self.cells
    for i, row in enumerate(cells):
        for j, _ in enumerate(row):
            mob = cells[i][j]
            mob.move_to(
                i * v_buff * DOWN + j * h_buff * RIGHT,
                element_alignment_corner,
            )
    self.elements = VGroup(*it.chain(*self.cells))
    """条目，逐行排列：以组的形式组织，并被添加到矩阵中。"""
    self.add(self.elements)
    bracket_config = {**bracket_config}
    BRACKET_HEIGHT = 0.5977
    n = int(self.height / BRACKET_HEIGHT) + 1
    empty_tex_array = "".join(
        ["\\begin{array}{c}", *n * ["\\quad \\\\"], "\\end{array}"]
    )
    tex_left = "".join(["\\left" + left_bracket, empty_tex_array, "\\right."])
    tex_right = "".join(["\\left.", empty_tex_array, "\\right" + right_bracket])
    l_bracket = MathTex(tex_left, **bracket_config)
    r_bracket = MathTex(tex_right, **bracket_config)
    bracket_pair = VGroup(l_bracket, r_bracket)
    if stretch_brackets:
        bracket_pair.stretch_to_fit_height(self.height + 2 * bracket_v_buff)
    l_bracket.next_to(self, LEFT, bracket_h_buff)
    r_bracket.next_to(self, RIGHT, bracket_h_buff)
    self.brackets = bracket_pair
    self.add(l_bracket, r_bracket)
    self.center()
    if add_background_rectangles_to_entries:
        for mob in self.elements:
            mob.add_background_rectangle()
    if include_background_rectangle:
        self.add_background_rectangle()

def __init__(
    self,
    table: Iterable[Iterable[float | str | Mobject]],
    row_labels: Iterable[Mobject] | None = None,
    col_labels: Iterable[Mobject] | None = None,
    top_left_entry: Mobject | None = None,
    v_buff: float = 0.8,
    h_buff: float = 1.3,
    include_outer_lines: bool = False,
    include_inner_lines: bool = True,
    add_background_rectangles_to_entries: bool = False,
    entries_background_color: ParsableManimColor = BLACK,
    include_background_rectangle: bool = False,
    background_rectangle_color: ParsableManimColor = BLACK,
    element_to_mobject: ElementFactory | None = None,
    element_to_mobject_config: Mapping[str, object] | None = None,
    arrange_in_grid_config: GridArrangement | None = None,
    line_config: LineOptions | None = None,
    **kwargs: Unpack[Style],
):
    if line_config is None:
        line_config = {}
    if arrange_in_grid_config is None:
        arrange_in_grid_config = {}
    table_data = [list(row) for row in table]
    row_label_data = list(row_labels) if row_labels is not None else []
    col_label_data = list(col_labels) if col_labels is not None else []
    self.row_labels = row_label_data or None
    """行标签，从上到下；若无则为 None。"""
    self.col_labels = col_label_data or None
    """列标签，从左到右；若无则为 None。"""
    self.top_left_entry = top_left_entry
    """标签之间角落处的 mobject；若无则为 None。"""
    self.row_dim = len(table_data)
    """表格的条目行数，不含标签。"""
    self.col_dim = len(table_data[0])
    """表格的条目列数，不含标签。"""
    self.v_buff = v_buff
    """各行之间的间隙，以场景单位表示，即表格创建时的值：缩放表格不会改变该值。"""
    self.h_buff = h_buff
    """各列之间的间隙，以场景单位表示，即表格创建时的值：缩放表格不会改变该值。"""
    entries_background_color = ManimColor(entries_background_color)
    background_rectangle_color = ManimColor(background_rectangle_color)
    if any(len(row) != len(table_data[0]) for row in table_data):
        raise ValueError("Not all rows in table have the same length.")
    super().__init__(**kwargs)
    cells = self._make_cells(
        table_data, element_to_mobject, element_to_mobject_config
    )
    self.elements_without_labels = VGroup(*it.chain(*cells))
    """不含标签的条目，逐行排列。"""
    empty_corner = None
    if self.row_labels is not None:
        for k in range(len(self.row_labels)):
            cells[k] = [self.row_labels[k]] + cells[k]
    if self.col_labels is not None:
        if self.row_labels is not None:
            if self.top_left_entry is not None:
                col_labels = [self.top_left_entry] + self.col_labels
                cells.insert(0, col_labels)
            else:
                empty_corner = VMobject()
                col_labels = [empty_corner] + self.col_labels
                cells.insert(0, col_labels)
        else:
            cells.insert(0, self.col_labels)
    self.cells = cells
    self.elements = VGroup(*it.chain(*self.cells))
    """表格的每个条目，包括标签，逐行排列：以组的形式组织，并被添加到表格中。"""
    layout = GridArrangement(
        rows=len(self.cells), cols=len(self.cells[0]), buff=(h_buff, v_buff)
    )
    self.elements.arrange_in_grid(**(layout | arrange_in_grid_config))
    if empty_corner is not None:
        self.elements.remove(empty_corner)
    self.add(self.elements)
    self.center()
    for axis in (0, 1):
        divisions = self.get_rows() if axis == 0 else self.get_columns()
        along, across = (RIGHT, UP) if axis == 0 else (DOWN, LEFT)
        near = [float(part.get_corner(across) @ across) for part in divisions]
        far = [float(part.get_corner(-across) @ across) for part in divisions]
        gap = (self.v_buff, self.h_buff)[axis] / 2
        pad = (self.h_buff, self.v_buff)[axis] / 2
        start = float(self.elements.get_corner(-along) @ along) - pad
        end = float(self.elements.get_corner(along) @ along) + pad
        anchors: list[tuple[float, bool]] = []
        if include_outer_lines:
            anchors.extend(((near[0] + gap, False), (far[-1] - gap, False)))
        if include_inner_lines:
            anchors.extend(
                (b + 0.5 * (a - b), axis == 1) for a, b in zip(far[:-1], near[1:])
            )
        lines = VGroup()
        parts: list[Line] = []
        for anchor, reverse in anchors:
            offset = anchor * across
            ends = (start * along + offset, end * along + offset)
            parts.append(Line(*(ends[::-1] if reverse else ends), **line_config))
        lines.add(*parts)
        self.add(*lines)
        if axis == 0:
            self.horizontal_lines = lines
        else:
            self.vertical_lines = lines
    if add_background_rectangles_to_entries:
        self.add_background_to_entries(color=entries_background_color)
    if include_background_rectangle:
        self.add_background_rectangle(color=background_rectangle_color)