Source code for qikify.models.helpers
def gz_csv_read(file_path, pandasDF=False):
[docs] import csv
import gzip
from StringIO import StringIO
with gzip.open(file_path, 'r') as f:
if pandasDF:
import pandas
data = pandas.read_csv(StringIO(f.read()))
else:
reader = csv.reader(StringIO(f.read()))
data = [row for row in reader]
return data
def gz_csv_write(file_path, data):
[docs] import csv
import gzip
from StringIO import StringIO
import pandas
with gzip.open(file_path, 'w') as f:
if isinstance(data, pandas.DataFrame):
data.to_csv(file_path, index=False)
else:
writer = csv.writer(f)
for row in data:
writer.writerow(row)