@@ -3668,15 +3668,15 @@ def get_object(self, obj, transposed: bool):
3668
3668
""" return the data for this obj """
3669
3669
return obj
3670
3670
3671
- def validate_data_columns (self , data_columns , min_itemsize ):
3671
+ def validate_data_columns (self , data_columns , min_itemsize , non_index_axes ):
3672
3672
"""take the input data_columns and min_itemize and create a data
3673
3673
columns spec
3674
3674
"""
3675
3675
3676
- if not len (self . non_index_axes ):
3676
+ if not len (non_index_axes ):
3677
3677
return []
3678
3678
3679
- axis , axis_labels = self . non_index_axes [0 ]
3679
+ axis , axis_labels = non_index_axes [0 ]
3680
3680
info = self .info .get (axis , dict ())
3681
3681
if info .get ("type" ) == "MultiIndex" and data_columns :
3682
3682
raise ValueError (
@@ -3835,7 +3835,9 @@ def get_blk_items(mgr, blocks):
3835
3835
blk_items = get_blk_items (block_obj ._data , blocks )
3836
3836
if len (new_non_index_axes ):
3837
3837
axis , axis_labels = new_non_index_axes [0 ]
3838
- data_columns = self .validate_data_columns (data_columns , min_itemsize )
3838
+ data_columns = self .validate_data_columns (
3839
+ data_columns , min_itemsize , new_non_index_axes
3840
+ )
3839
3841
if len (data_columns ):
3840
3842
mgr = block_obj .reindex (
3841
3843
Index (axis_labels ).difference (Index (data_columns )), axis = axis
@@ -3931,6 +3933,7 @@ def get_blk_items(mgr, blocks):
3931
3933
self .data_columns = new_data_columns
3932
3934
self .values_axes = vaxes
3933
3935
self .index_axes = new_index_axes
3936
+ self .non_index_axes = new_non_index_axes
3934
3937
3935
3938
# validate our min_itemsize
3936
3939
self .validate_min_itemsize (min_itemsize )
0 commit comments