catch zip

Author: minggli

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