Skip to content

Commit 9c5165e

Browse files
mahepejreback
authored andcommitted
BUG: Fix MultiIndex DataFrame to_csv() segfault (#26303) (#26355)
1 parent 80e2615 commit 9c5165e

File tree

3 files changed

+13
-2
lines changed

3 files changed

+13
-2
lines changed

doc/source/whatsnew/v0.25.0.rst

+1-1
Original file line numberDiff line numberDiff line change
@@ -373,7 +373,7 @@ Indexing
373373
- Improved exception message when calling :meth:`DataFrame.iloc` with a list of non-numeric objects (:issue:`25753`).
374374
- Bug in :meth:`DataFrame.loc` and :meth:`Series.loc` where ``KeyError`` was not raised for a ``MultiIndex`` when the key was less than or equal to the number of levels in the :class:`MultiIndex` (:issue:`14885`).
375375
- Bug in which :meth:`DataFrame.append` produced an erroneous warning indicating that a ``KeyError`` will be thrown in the future when the data to be appended contains new columns (:issue:`22252`).
376-
-
376+
- Bug in which :meth:`DataFrame.to_csv` caused a segfault for a reindexed data frame, when the indices were single-level :class:`MultiIndex` (:issue:`26303`).
377377

378378

379379
Missing

pandas/core/indexes/multi.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -946,7 +946,9 @@ def _format_native_types(self, na_rep='nan', **kwargs):
946946
new_codes.append(level_codes)
947947

948948
if len(new_levels) == 1:
949-
return Index(new_levels[0])._format_native_types()
949+
# a single-level multi-index
950+
return Index(new_levels[0].take(
951+
new_codes[0]))._format_native_types()
950952
else:
951953
# reconstruct the multi-index
952954
mi = MultiIndex(levels=new_levels, codes=new_codes,

pandas/tests/frame/test_to_csv.py

+9
Original file line numberDiff line numberDiff line change
@@ -1230,6 +1230,15 @@ def test_multi_index_header(self):
12301230
expected = tm.convert_rows_list_to_csv_str(expected_rows)
12311231
assert result == expected
12321232

1233+
def test_to_csv_single_level_multi_index(self):
1234+
# see gh-26303
1235+
index = pd.Index([(1,), (2,), (3,)])
1236+
df = pd.DataFrame([[1, 2, 3]], columns=index)
1237+
df = df.reindex(columns=[(1,), (3,)])
1238+
expected = ",1,3\n0,1,3\n"
1239+
result = df.to_csv(line_terminator='\n')
1240+
assert_almost_equal(result, expected)
1241+
12331242
def test_gz_lineend(self):
12341243
# GH 25311
12351244
df = pd.DataFrame({'a': [1, 2]})

0 commit comments

Comments
 (0)