Adding repace_table()

pptx_blueprint/__init__.py

@@ -42,14 +42,57 @@ def replace_picture(self, label: str, filename: _Pathlike) -> None:
         """
         pass
 
-    def replace_table(self, label: str, data) -> None:
+    def replace_table(self, label: str, data, header=False, rownames=False):
         """Replaces rectangle placeholders on one or many slides.
 
         Args:
             label (str): label of the placeholder (without curly braces)
             data (pandas.DataFrame): table to be inserted into the presentation
         """
-        pass
+        assert isinstance(data, pandas.dataframe)
+
+        shapes_to_replace = self._find_shapes(label)
+
+        rows, cols = data.shape
+        if header: rows += 1
+        if rownames: cols += 1
+
+        for old_shape in shapes_to_replace:
+            slide_shapes = old_shape._parent
+            table = slide_shapes.add_table(
+                rows,
+                cols,
+                old_shape.left, 
+                old_shape.top, 
+                old_shape.width,
+                old_shape.height
+            ).table
+
+            # set column widths
+            col_width = Length(old_shape.width/cols)
+            for i in range(len(table.columns)):
+                table.columns[i].width = col_width
+
+            ## fill the table
+            for c in range(data.shape[1]):
+                for r in range(data.shape[0]):
+                    if header and r == 0:
+                        c_temp = c
+                        ## when rownames, skip first column
+                        if rownames: c_temp += 1
+                        table.cell(r, c_temp).text = data.columns[c]
+                    if rownames and c == 0:
+                        r_temp = r
+                        ## when header, skip first row
+                        if header: r_temp += 1
+                        table.cell(r_temp, c).text = str(data.index[r])
+                    ## fill table body
+                    r_shape = r
+                    if header: r_shape += 1
+                    c_shape = c
+                    if rownames: c_shape += 1
+                    table.cell(r_shape, c_shape).text = str(data.iloc[r, c])
+
 
     def _parse_label(self, label: str) -> Tuple[Union[int, str], str]:
         slide_number, tag_name = label.split(':')