File tree 1 file changed +6
-11
lines changed
1 file changed +6
-11
lines changed Original file line number Diff line number Diff line change 2
2
concat routines
3
3
"""
4
4
5
- from typing import List
5
+ from typing import Hashable , List , Optional
6
6
7
7
import numpy as np
8
8
@@ -474,15 +474,10 @@ def _get_result_dim(self) -> int:
474
474
475
475
def _get_new_axes (self ) -> List [Index ]:
476
476
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
+ ]
486
481
487
482
def _get_comb_axis (self , i : int ) -> Index :
488
483
data_axis = self .objs [0 ]._get_block_manager_axis (i )
@@ -501,7 +496,7 @@ def _get_concat_axis(self) -> Index:
501
496
idx = ibase .default_index (len (self .objs ))
502
497
return idx
503
498
elif self .keys is None :
504
- names : List = [None ] * len (self .objs )
499
+ names : List [ Optional [ Hashable ]] = [None ] * len (self .objs )
505
500
num = 0
506
501
has_names = False
507
502
for i , x in enumerate (self .objs ):
You can’t perform that action at this time.
0 commit comments