hiframes df.col.values

Author: Ehsan Totoni

hpat/hiframes.py

@@ -140,6 +140,14 @@ def _run_assign(self, assign):
                 assign.value = df_cols[rhs.attr]
                 self.df_cols.add(lhs)  # save lhs as column
 
+            # c = df.column.values
+            if (rhs.op=='getattr' and rhs.value.name in self.df_cols and
+                        rhs.attr == 'values'):
+                # simply return the column
+                # output is array so it's not added to df_cols
+                assign.value = rhs.value
+                return [assign]
+
             # c = df.column.shift
             if (rhs.op=='getattr' and rhs.value.name in self.df_cols and
                         rhs.attr in df_col_funcs):

hpat/tests/test_hiframes.py

@@ -6,6 +6,19 @@
                             count_parfor_OneDs, count_array_OneDs, dist_IR_contains)
 
 class TestHiFrames(unittest.TestCase):
+    def test_basics(self):
+        def test_impl(n):
+            df = pd.DataFrame({'A': np.ones(n), 'B': np.random.ranf(n)})
+            Ac = df['A'].values
+            return Ac.sum()
+
+        hpat_func = hpat.jit(test_impl)
+        n = 11
+        self.assertEqual(hpat_func(n), test_impl(n))
+        self.assertEqual(count_array_REPs(), 0)
+        self.assertEqual(count_parfor_REPs(), 0)
+        self.assertEqual(count_parfor_OneDs(), 1)
+
     def test_cumsum(self):
         def test_impl(n):
             df = pd.DataFrame({'A': np.ones(n), 'B': np.random.ranf(n)})