Skip to content

Commit 984a2ca

Browse files
simonjayhawkinsWillAyd
authored andcommitted
TYP: --disallow-any-generics pandas\core\reshape\concat.py (#30658)
1 parent c1b8573 commit 984a2ca

File tree

1 file changed

+6
-11
lines changed

1 file changed

+6
-11
lines changed

pandas/core/reshape/concat.py

+6-11
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
concat routines
33
"""
44

5-
from typing import List
5+
from typing import Hashable, List, Optional
66

77
import numpy as np
88

@@ -474,15 +474,10 @@ def _get_result_dim(self) -> int:
474474

475475
def _get_new_axes(self) -> List[Index]:
476476
ndim = self._get_result_dim()
477-
new_axes: List = [None] * ndim
478-
479-
for i in range(ndim):
480-
if i == self.axis:
481-
continue
482-
new_axes[i] = self._get_comb_axis(i)
483-
484-
new_axes[self.axis] = self._get_concat_axis()
485-
return new_axes
477+
return [
478+
self._get_concat_axis() if i == self.axis else self._get_comb_axis(i)
479+
for i in range(ndim)
480+
]
486481

487482
def _get_comb_axis(self, i: int) -> Index:
488483
data_axis = self.objs[0]._get_block_manager_axis(i)
@@ -501,7 +496,7 @@ def _get_concat_axis(self) -> Index:
501496
idx = ibase.default_index(len(self.objs))
502497
return idx
503498
elif self.keys is None:
504-
names: List = [None] * len(self.objs)
499+
names: List[Optional[Hashable]] = [None] * len(self.objs)
505500
num = 0
506501
has_names = False
507502
for i, x in enumerate(self.objs):

0 commit comments

Comments
 (0)