Prechádzať zdrojové kódy

adding queries to files

Steve Nyemba 3 mesiacov pred
rodič
commit
a4597d4a8c
1 zmenil súbory, kde vykonal 5 pridanie a 1 odobranie
  1. 5 1
      transport/other/files.py

+ 5 - 1
transport/other/files.py

@@ -30,7 +30,11 @@ class Reader (File):
 	def read(self,**args):
 		_path = self.path if 'path' not in args else args['path']
 		_delimiter = self.delimiter if 'delimiter' not in args else args['delimiter']
-		return pd.read_csv(_path,delimiter=self.delimiter)
+		_df =  pd.read_csv(_path,delimiter=self.delimiter)
+		if 'query' in args :
+			_query = args['query']
+			_df = _df.query(_query)
+		return _df
 	def stream(self,**args):
 		raise Exception ("streaming needs to be implemented")
 class Writer (File):