Add csv_to_df function

pydanny · Nov 27, 2023 · c2cd709 · c2cd709
1 parent c545c64
commit c2cd709
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/dj_notebook/shell_plus.py b/src/dj_notebook/shell_plus.py
@@ -13,6 +13,8 @@
 
 
 import base64
+import io
+import pathlib
 import typing
 
 import IPython
@@ -143,6 +145,14 @@ def model_graph(self, model: django_models.Model, max_nodes: int = 20) -> None:
             )
         display_mermaid(output)
 
+    def csv_to_df(self, filepath_or_string: pathlib.Path | str) -> pd.DataFrame:
+        """Read a CSV file into a Pandas DataFrame."""
+        if isinstance(filepath_or_string, pathlib.Path):
+            return pd.read_csv(filepath_or_string)
+        buffer = io.StringIO(filepath_or_string)        
+        return pd.read_csv(buffer)
+
+
 
 def get_node_for_model(graph, model: django_models.Model):
     try: