from_pandas: add index option

rxsci/operators/pandas.py

@@ -10,7 +10,7 @@
     except Exception:
         pass
 
-    def from_pandas(dataframe, scheduler=None, progress=False):
+    def from_pandas(dataframe, scheduler=None, progress=False, index=False):
         """Creates an observable from a pandas dataframe
 
         When a dict is provided as the progress argument, it accepts these keys:
@@ -31,11 +31,11 @@ def from_pandas(dataframe, scheduler=None, progress=False):
                 tqdm_kwargs['total']=len(dataframe)
 
             return rx.from_(
-                tqdm(dataframe.itertuples(index=False), **tqdm_kwargs),
+                tqdm(dataframe.itertuples(index=index), **tqdm_kwargs),
                 scheduler=scheduler,
             )
         return rx.from_(
-            dataframe.itertuples(index=False),
+            dataframe.itertuples(index=index),
             scheduler=scheduler,
         )
 

tests/operators/test_pandas.py

@@ -33,6 +33,28 @@ def test_from_pandas_base(progress):
     assert actual_result == 3
 
 
+@pytest.mark.parametrize(
+    "progress",
+    [
+        False, True, {'interval': 60}
+    ]
+)
+def test_from_pandas_with_index(progress):
+    data = [
+        Item("foo", 1),
+        Item("bar", 2),
+        Item("fiz", 4),
+        Item("fuz", 4),
+    ]
+
+    df = pd.DataFrame(data)
+    actual_result = rs.ops.from_pandas(df, progress=progress, index=True).pipe(
+        ops.map(lambda i: i.Index),
+        ops.sum(),
+    ).run()
+
+    assert actual_result == 6
+
 def test_to_pandas_base():
     data = [
         Item("foo", 1),