(issue 805) updated LArray.to_frame(): call AxisCollection.display_names property at the beginning to to make LArray.to_frame() consistent with LArray.dump() (and then to make it possible to handle anonymous and/or wildcard axes when dealing with CSV and Excel formats)

Author: alixdamman

larray/core/array.py

@@ -1185,22 +1185,21 @@ def to_frame(self, fold_last_axis_name=False, dropna=None):
            b1    6   7
         """
         columns = pd.Index(self.axes[-1].labels)
+        axes_names = self.axes.display_names[:]
         if not fold_last_axis_name:
-            columns.name = self.axes[-1].name
+            columns.name = axes_names[-1]
         if self.ndim > 1:
-            axes_names = self.axes.names[:-1]
+            _axes_names = axes_names[:-1]
             if fold_last_axis_name:
-                tmp = axes_names[-1] if axes_names[-1] is not None else ''
-                if self.axes[-1].name:
-                    axes_names[-1] = "{}\\{}".format(tmp, self.axes[-1].name)
+                _axes_names[-1] = "{}\\{}".format(_axes_names[-1], axes_names[-1])
             if self.ndim == 2:
-                index = pd.Index(data=self.axes[0].labels, name=axes_names[0])
+                index = pd.Index(data=self.axes[0].labels, name=_axes_names[0])
             else:
-                index = pd.MultiIndex.from_product(self.axes.labels[:-1], names=axes_names)
+                index = pd.MultiIndex.from_product(self.axes.labels[:-1], names=_axes_names)
         else:
             index = pd.Index([''])
             if fold_last_axis_name:
-                index.name = self.axes.names[-1]
+                index.name = axes_names[-1]
         data = np.asarray(self).reshape(len(index), len(columns))
         df = pd.DataFrame(data, index, columns)
         if dropna is not None:

larray/tests/test_array.py

@@ -3454,6 +3454,17 @@ def test_from_series():
     assert_array_equal(res, expected)
 
 
+def test_to_frame():
+    # array containing anonymous axes
+    arr = ndtest((Axis(2), Axis(2), Axis(2)))
+    df = arr.to_frame()
+    assert df.index.name is None
+    assert df.index.names == ['{0}*', '{1}*']
+    assert df.columns.name == '{2}*'
+    assert list(df.index.values) == [(0, 0), (0, 1), (1, 0), (1, 1)]
+    assert list(df.columns.values) == [0, 1]
+
+
 def test_from_frame():
     # 1) data = scalar
     # ================