Skip to content

Commit 1905fcc

Browse files
committed
enhancement to dataframe to_csv zip method,add this step to perform like gz or bz2
1 parent d558bce commit 1905fcc

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

pandas/io/common.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -726,6 +726,17 @@ def __init__(
726726
self.archive_name = archive_name
727727
self.multiple_write_buffer: Optional[Union[StringIO, BytesIO]] = None
728728

729+
# add csv file name like gz or bz2
730+
try:
731+
fname = os.path.basename(file)
732+
if not isinstance(fname, bytes):
733+
fname = fname.encode('latin-1')
734+
if fname.endswith(b'.zip'):
735+
xname = fname[:-4].decode('latin-1')
736+
self.archive_name=xname
737+
except:
738+
pass
739+
729740
kwargs_zip: Dict[str, Any] = {"compression": zipfile.ZIP_DEFLATED}
730741
kwargs_zip.update(kwargs)
731742

0 commit comments

Comments
 (0)