def import_set(dset, in_stream, headers=True): """Returns dataset from CSV stream.""" dset.wipe() if is_py3: rows = csv.reader(StringIO(in_stream)) else: rows = csv.reader(StringIO(in_stream), encoding=DEFAULT_ENCODING) for i, row in enumerate(rows): if (i == 0) and (headers): dset.headers = row else: dset.append(row)
def import_set(dset, in_stream, headers=True): """Returns dataset from CSV stream.""" dset.wipe() if is_py3: rows = csv.reader(in_stream.splitlines()) else: rows = csv.reader(in_stream.splitlines(), encoding=DEFAULT_ENCODING) for i, row in enumerate(rows): if (i == 0) and (headers): dset.headers = row else: dset.append(row)
def import_set(dset, in_stream, headers=True): """Returns dataset from TSV stream.""" dset.wipe() if is_py3: rows = csv.reader(in_stream.split('\r\n'), delimiter='\t') else: rows = csv.reader(in_stream.split('\r\n'), delimiter='\t', encoding=DEFAULT_ENCODING) for i, row in enumerate(rows): # Skip empty rows if not row: continue if (i == 0) and (headers): dset.headers = row else: dset.append(row)
def import_set(dset, in_stream, headers=True, **kwargs): """Returns dataset from CSV stream.""" dset.wipe() kwargs.setdefault('delimiter', DEFAULT_DELIMITER) rows = csv.reader(StringIO(in_stream), **kwargs) for i, row in enumerate(rows): if (i == 0) and (headers): dset.headers = row elif row: dset.append(row)