From 40d332154b5d45f19666e8e834878d8563cfc9a4 Mon Sep 17 00:00:00 2001 From: Simon Hawkins Date: Fri, 3 Jan 2020 20:19:31 +0000 Subject: [PATCH] TYP: --disallow-any-generics pandas\core\reshape\concat.py --- pandas/core/reshape/concat.py | 17 ++++++----------- 1 file changed, 6 insertions(+), 11 deletions(-) diff --git a/pandas/core/reshape/concat.py b/pandas/core/reshape/concat.py index a3d9dbfba9e71..1414dbeb9b950 100644 --- a/pandas/core/reshape/concat.py +++ b/pandas/core/reshape/concat.py @@ -2,7 +2,7 @@ concat routines """ -from typing import List +from typing import Hashable, List, Optional import numpy as np @@ -474,15 +474,10 @@ def _get_result_dim(self) -> int: def _get_new_axes(self) -> List[Index]: ndim = self._get_result_dim() - new_axes: List = [None] * ndim - - for i in range(ndim): - if i == self.axis: - continue - new_axes[i] = self._get_comb_axis(i) - - new_axes[self.axis] = self._get_concat_axis() - return new_axes + return [ + self._get_concat_axis() if i == self.axis else self._get_comb_axis(i) + for i in range(ndim) + ] def _get_comb_axis(self, i: int) -> Index: data_axis = self.objs[0]._get_block_manager_axis(i) @@ -501,7 +496,7 @@ def _get_concat_axis(self) -> Index: idx = ibase.default_index(len(self.objs)) return idx elif self.keys is None: - names: List = [None] * len(self.objs) + names: List[Optional[Hashable]] = [None] * len(self.objs) num = 0 has_names = False for i, x in enumerate(self.objs):