Skip to content

Commit 417e74a

Browse files
committed
FIX: fix index labels
1 parent cdeb115 commit 417e74a

File tree

1 file changed

+12
-11
lines changed

1 file changed

+12
-11
lines changed

pandas/io/formats/csvs.py

+12-11
Original file line numberDiff line numberDiff line change
@@ -98,16 +98,20 @@ def index_label(self):
9898

9999
@index_label.setter
100100
def index_label(self, index_label) -> None:
101-
if index_label is None:
102-
self._index_label = self._get_index_label_from_obj()
103-
elif not isinstance(index_label, (list, tuple, np.ndarray, ABCIndexClass)):
104-
# given a string for a DF with Index
105-
self._index_label = [index_label]
101+
# should write something for index label
102+
if index_label is not False:
103+
if index_label is None:
104+
index_label = self._get_index_label_from_obj()
105+
elif not isinstance(index_label, (list, tuple, np.ndarray, ABCIndexClass)):
106+
# given a string for a DF with Index
107+
index_label = [index_label]
108+
self._index_label = index_label
106109

107110
def _get_index_label_from_obj(self) -> List[str]:
108111
if isinstance(self.obj.index, ABCMultiIndex):
109112
return self._get_index_label_multiindex()
110-
return self._get_index_label_regular()
113+
else:
114+
return self._get_index_label_regular()
111115

112116
def _get_index_label_multiindex(self) -> List[str]:
113117
return [name or "" for name in self.obj.index.names]
@@ -189,10 +193,7 @@ def data_index(self):
189193
from pandas import Index
190194

191195
data_index = Index(
192-
[
193-
x.strftime(self.date_format) if notna(x) else ""
194-
for x in self.data_index
195-
]
196+
[x.strftime(self.date_format) if notna(x) else "" for x in data_index]
196197
)
197198
return data_index
198199

@@ -249,7 +250,7 @@ def write_cols(self):
249250
def encoded_labels(self) -> List[str]:
250251
encoded_labels: List[str] = []
251252

252-
if self.index:
253+
if self.index and self.index_label:
253254
encoded_labels = list(self.index_label)
254255

255256
if not self.has_mi_columns or self._has_aliases:

0 commit comments

Comments
 (0)