Skip to content

Commit f6f6bfa

Browse files
simplify code a bit
1 parent 8492e1e commit f6f6bfa

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

pandas/io/formats/csvs.py

+4-5
Original file line numberDiff line numberDiff line change
@@ -213,21 +213,20 @@ def _index_label_encoder(self):
213213
obj = self.obj
214214

215215
if index_label is True:
216+
index_label = []
216217
# append index label based on index type
217218
if isinstance(obj.index, ABCMultiIndex):
218-
index_label = []
219219
for name in obj.index.names:
220220
# add empty string is name is None
221221
if name is None:
222222
name = ''
223223
index_label.append(name)
224224
else:
225-
index_label = obj.index.name
226225
# if no name, use empty string
227-
if index_label is None:
228-
index_label = ['']
226+
if obj.index.name is None:
227+
index_label.append('')
229228
else:
230-
index_label = [index_label]
229+
index_label.append(obj.index.name)
231230
elif not isinstance(index_label,
232231
(list, tuple, np.ndarray, ABCIndexClass)):
233232
index_label = [index_label]

0 commit comments

Comments
 (0)