ENH: add index param to to_dict method pandas-dev#46398

Author: Vruthik Thakkar

pandas/core/frame.py

@@ -1771,7 +1771,7 @@ def to_numpy(
 
         return result
 
-    def to_dict(self, orient: str = "dict", into=dict):
+    def to_dict(self, orient: str = "dict", into=dict, index=True):
         """
         Convert the DataFrame to a dictionary.
 
@@ -1807,6 +1807,11 @@ def to_dict(self, orient: str = "dict", into=dict):
             instance of the mapping type you want.  If you want a
             collections.defaultdict, you must pass it initialized.
 
+        index : bool default True
+            When set to False, method returns a dict without an index key
+            (and index_names if using orient='tight'). Can only be False
+            when orient='split' or 'tight'.
+
         Returns
         -------
         dict, list or collections.abc.Mapping
@@ -1909,43 +1914,77 @@ def to_dict(self, orient: str = "dict", into=dict):
             elif orient.startswith("i"):
                 orient = "index"
 
+        if not index and orient not in ['split', 'tight']:
+            raise ValueError(
+                "'index=False' is only valid when 'orient' is 'split' or 'tight"
+            )
+
         if orient == "dict":
             return into_c((k, v.to_dict(into)) for k, v in self.items())
 
         elif orient == "list":
             return into_c((k, v.tolist()) for k, v in self.items())
 
         elif orient == "split":
-            return into_c(
-                (
-                    ("index", self.index.tolist()),
-                    ("columns", self.columns.tolist()),
+            if not index:
+                return into_c(
                     (
-                        "data",
-                        [
-                            list(map(maybe_box_native, t))
-                            for t in self.itertuples(index=False, name=None)
-                        ],
-                    ),
+                        ("columns", self.columns.tolist()),
+                        (
+                            "data",
+                            [
+                                list(map(maybe_box_native, t))
+                                for t in self.itertuples(index=False, name=None)
+                            ],
+                        ),
+                    )
+                )
+            else:
+                return into_c(
+                    (
+                        ("index", self.index.tolist()),
+                        ("columns", self.columns.tolist()),
+                        (
+                            "data",
+                            [
+                                list(map(maybe_box_native, t))
+                                for t in self.itertuples(index=False, name=None)
+                            ],
+                        ),
+                    )
                 )
-            )
 
         elif orient == "tight":
-            return into_c(
-                (
-                    ("index", self.index.tolist()),
-                    ("columns", self.columns.tolist()),
+            if not index:
+                return into_c(
+                    (
+                        ("columns", self.columns.tolist()),
+                        (
+                            "data",
+                            [
+                                list(map(maybe_box_native, t))
+                                for t in self.itertuples(index=False, name=None)
+                            ],
+                        ),
+                        ("column_names", list(self.columns.names)),
+                    )
+                )
+            else:
+                return into_c(
                     (
-                        "data",
-                        [
-                            list(map(maybe_box_native, t))
-                            for t in self.itertuples(index=False, name=None)
-                        ],
-                    ),
-                    ("index_names", list(self.index.names)),
-                    ("column_names", list(self.columns.names)),
+                        ("index", self.index.tolist()),
+                        ("columns", self.columns.tolist()),
+                        (
+                            "data",
+                            [
+                                list(map(maybe_box_native, t))
+                                for t in self.itertuples(index=False, name=None)
+                            ],
+                        ),
+                        ("index_names", list(self.index.names)),
+                        ("column_names", list(self.columns.names)),
+                    )
                 )
-            )
 
         elif orient == "series":
             return into_c((k, v) for k, v in self.items())