在Python中對gzip文件使用csvreader

Question

我有一堆gzip的CSV文件，我想用Python內置的CSV閱讀器打開以供檢查。我想這樣做，而不必先手動將它們解壓縮到磁盤。我想我想以某種方式獲得未壓縮數據的流，並將其傳遞到CSV閱讀器。這在Python中可能嗎？

Answer 1

使用gzip模塊：

with gzip.open(filename) as f: 
    reader = csv.reader(f) 
    #... 

Answer 2

一個更完整的解決方案：

import csv, gzip 
class GZipCSVReader: 
    def __init__(self, filename): 
     self.gzfile = gzip.open(filename) 
     self.reader = csv.DictReader(self.gzfile) 
    def next(self): 
     return self.reader.next() 
    def close(self): 
     self.gzfile.close() 
    def __iter__(self): 
     return self.reader.__iter__() 

現在你可以使用它像這樣：

r = GZipCSVReader('my.csv') 
for map in r: 
    for k,v in map: 
     print k,v 
r.close() 

Answer 3

我已經試過以上版本用於寫入和閱讀，並且由於「字節」錯誤，它在Python 3.3中不起作用。但是，經過一些試驗和錯誤，我可以得到以下工作。也許這也幫助他人：

import csv 
import gzip 
import io 


with gzip.open("test.gz", "w") as file: 
    writer = csv.writer(io.TextIOWrapper(file, newline="", write_through=True)) 
    writer.writerow([1, 2, 3]) 
    writer.writerow([4, 5, 6]) 

with gzip.open("test.gz", "r") as file: 
    reader = csv.reader(io.TextIOWrapper(file, newline="")) 
    print(list(reader)) 

由於amohr建議，以下的作品，以及：

import gzip, csv 

with gzip.open("test.gz", "wt", newline="") as file: 
    writer = csv.writer(file) 
    writer.writerow([1, 2, 3]) 
    writer.writerow([4, 5, 6]) 

with gzip.open("test.gz", "rt", newline="") as file: 
    reader = csv.reader(file) 
    print(list(reader))