BUG: GH #12223, GH #15262. Allow ints for names in MultiIndex

Author: Dr-Irv

doc/source/whatsnew/v0.20.0.txt

@@ -153,6 +153,7 @@ Other enhancements
 - ``Series/DataFrame.resample.asfreq`` have gained a ``fill_value`` parameter, to fill missing values during resampling (:issue:`3715`).
 - ``pandas.tools.hashing`` has gained a ``hash_tuples`` routine, and ``hash_pandas_object`` has gained the ability to hash a ``MultiIndex`` (:issue:`15224`)
 - ``Series/DataFrame.squeeze()`` have gained the ``axis`` parameter. (:issue:`15339`)
+- Using numerical names in ``MultiIndex`` causes less errors. (:issue:`12223`) (:issue:`15262`)
 
 .. _ISO 8601 duration: https://en.wikipedia.org/wiki/ISO_8601#Durations
 

pandas/core/frame.py

@@ -2860,7 +2860,7 @@ def set_index(self, keys, drop=True, append=False, inplace=False,
             names = [x for x in self.index.names]
             if isinstance(self.index, MultiIndex):
                 for i in range(self.index.nlevels):
-                    arrays.append(self.index.get_level_values(i))
+                    arrays.append(self.index._get_level_values_by_level(i))
             else:
                 arrays.append(self.index)
 
@@ -2870,9 +2870,9 @@ def set_index(self, keys, drop=True, append=False, inplace=False,
                 # append all but the last column so we don't have to modify
                 # the end of this loop
                 for n in range(col.nlevels - 1):
-                    arrays.append(col.get_level_values(n))
+                    arrays.append(col._get_level_values_by_level(n))
 
-                level = col.get_level_values(col.nlevels - 1)
+                level = col._get_level_values_by_level(col.nlevels - 1)
                 names.extend(col.names)
             elif isinstance(col, Series):
                 level = col._values

pandas/core/groupby.py

@@ -289,7 +289,7 @@ def _set_grouper(self, obj, sort=False):
                 # equivalent to the axis name
                 if isinstance(ax, MultiIndex):
                     level = ax._get_level_number(level)
-                    ax = Index(ax.get_level_values(
+                    ax = Index(ax._get_level_values_by_level(
                         level), name=ax.names[level])
 
                 else:
@@ -759,7 +759,7 @@ def _index_with_as_index(self, b):
         gp = self.grouper
         levels = chain((gp.levels[i][gp.labels[i][b]]
                         for i in range(len(gp.groupings))),
-                       (original.get_level_values(i)[b]
+                       (original._get_level_values_by_level(i)[b]
                         for i in range(original.nlevels)))
         new = MultiIndex.from_arrays(list(levels))
         new.names = gp.names + original.names

pandas/core/reshape.py

@@ -814,7 +814,8 @@ def melt(frame, id_vars=None, value_vars=None, var_name=None,
     mdata[value_name] = frame.values.ravel('F')
     for i, col in enumerate(var_name):
         # asanyarray will keep the columns as an Index
-        mdata[col] = np.asanyarray(frame.columns.get_level_values(i)).repeat(N)
+        mdata[col] = np.asanyarray(frame.columns
+                                   ._get_level_values_by_level(i)).repeat(N)
 
     return DataFrame(mdata, columns=mcolumns)
 

pandas/formats/format.py

@@ -1566,7 +1566,7 @@ def _save_header(self):
                     if isinstance(index_label, list) and len(index_label) > 1:
                         col_line.extend([''] * (len(index_label) - 1))
 
-                col_line.extend(columns.get_level_values(i))
+                col_line.extend(columns._get_level_values_by_level(i))
 
                 writer.writerow(col_line)
 

pandas/indexes/base.py

@@ -2346,6 +2346,11 @@ def get_level_values(self, level):
         self._validate_index_level(level)
         return self
 
+    def _get_level_values_by_level(self, num):
+        # Used to mirror implementation for MultiIndex
+        # GH #10461
+        return self.get_level_values(num)
+
     def get_indexer(self, target, method=None, limit=None, tolerance=None):
         """
         Compute indexer and mask for new index given the current index. The

pandas/indexes/multi.py

@@ -761,7 +761,24 @@ def get_level_values(self, level):
         -------
         values : ndarray
         """
-        num = self._get_level_number(level)
+        return self._get_level_values_by_level(self._get_level_number(level))
+
+    def _get_level_values_by_level(self, num):
+        """
+        Return vector of label values for requested level, equal to the length
+        of the index. Assumes a level number.  For internal use
+
+        Parameters
+        ----------
+        num : int
+
+        Returns
+        -------
+        values : ndarray
+        """
+        # Discussion in GH 10461
+        # When you know the parameter is a level number, use this
+        # Provides better support when names in a MultiIndex are ints
         unique = self.levels[num]  # .values
         labels = self.labels[num]
         filled = algos.take_1d(unique.values, labels,
@@ -850,7 +867,8 @@ def to_frame(self, index=True):
         """
 
         from pandas import DataFrame
-        result = DataFrame({(name or level): self.get_level_values(level)
+        result = DataFrame({(name or level):
+                            self._get_level_values_by_level(level)
                             for name, level in
                             zip(self.names, range(len(self.levels)))})
         if index:
@@ -1184,8 +1202,8 @@ def append(self, other):
                for o in other):
             arrays = []
             for i in range(self.nlevels):
-                label = self.get_level_values(i)
-                appended = [o.get_level_values(i) for o in other]
+                label = self._get_level_values_by_level(i)
+                appended = [o._get_level_values_by_level(i) for o in other]
                 arrays.append(label.append(appended))
             return MultiIndex.from_arrays(arrays, names=self.names)
 

pandas/io/sql.py

@@ -748,7 +748,7 @@ def _get_column_names_and_types(self, dtype_mapper):
         if self.index is not None:
             for i, idx_label in enumerate(self.index):
                 idx_type = dtype_mapper(
-                    self.frame.index.get_level_values(i))
+                    self.frame.index._get_level_values_by_level(i))
                 column_names_and_types.append((idx_label, idx_type, True))
 
         column_names_and_types += [

pandas/tests/frame/test_combine_concat.py

@@ -422,6 +422,23 @@ def test_concat_axis_parameter(self):
         with assertRaisesRegexp(ValueError, 'No axis named'):
             pd.concat([series1, series2], axis='something')
 
+    def test_concat_numerical_names(self):
+        # #15262  # #12223
+        df = pd.DataFrame({'col': range(9)},
+                          index=(pd.MultiIndex
+                                 .from_product([['A0', 'A1', 'A2'],
+                                                ['B0', 'B1', 'B2']],
+                                               names=[1, 2])))
+        result = pd.concat((df.iloc[:2, :], df.iloc[-2:, :]))
+        expected = pd.DataFrame({'col': [0, 1, 7, 8]},
+                                dtype='int32',
+                                index=pd.MultiIndex.from_tuples([('A0', 'B0'),
+                                                                 ('A0', 'B1'),
+                                                                 ('A2', 'B1'),
+                                                                 ('A2', 'B2')],
+                                                                names=[1, 2]))
+        tm.assert_frame_equal(result, expected)
+
 
 class TestDataFrameCombineFirst(tm.TestCase, TestData):
 

pandas/util/doctools.py

@@ -113,12 +113,12 @@ def _insert_index(self, data):
         else:
             for i in range(idx_nlevels):
                 data.insert(i, 'Index{0}'.format(i),
-                            data.index.get_level_values(i))
+                            data.index._get_level_values_by_level(i))
 
         col_nlevels = data.columns.nlevels
         if col_nlevels > 1:
-            col = data.columns.get_level_values(0)
-            values = [data.columns.get_level_values(i).values
+            col = data.columns._get_level_values_by_level(0)
+            values = [data.columns._get_level_values_by_level(i).values
                       for i in range(1, col_nlevels)]
             col_df = pd.DataFrame(values)
             data.columns = col_df.columns