catch zip

pandas-dev · Jun 14, 2018 · 0da6c41 · 0da6c41
1 parent 6accf04
commit 0da6c41
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pandas/io/formats/csvs.py b/pandas/io/formats/csvs.py
@@ -6,6 +6,7 @@
 from __future__ import print_function
 
 import csv as csvlib
+from zipfile import ZipFile
 import numpy as np
 
 from pandas.core.dtypes.missing import notna
@@ -127,8 +128,9 @@ def save(self):
         else:
             encoding = self.encoding
 
-        if (not hasattr(self.path_or_buf, 'write') and
-                self.compression == 'zip'):
+        if (isinstance(self.path_or_buf, ZipFile) or
+            (not hasattr(self.path_or_buf, 'write') and
+                self.compression == 'zip')):
             is_zip = True
         else:
             is_zip = False