def as_dataframe(self, chunk_size = None): try: from pandas import DataFrame except: raise Exception("pandas required to select dataframe. " "Please install: sudo pip install pandas") if chunk_size: rows = self.cursor.fetchN(chunk_size) data = map(self._parse_row, rows) return list_to_dataframe(data, self.columns) return list_to_dataframe(self.data, self.columns)
def as_dataframe(self, chunk_size=None): try: from pandas import DataFrame except: raise Exception("pandas required to select dataframe. " "Please install: sudo pip install pandas") if chunk_size: rows = self.cursor.fetchN(chunk_size) data = map(self._parse_row, rows) return list_to_dataframe(data, self.columns) return list_to_dataframe(self.data, self.columns)
def as_dataframe(self): try: from pandas import DataFrame except: raise Exception("pandas required to select dataframe. " "Please install: sudo pip install pandas") return list_to_dataframe(self.data, self.columns)
def as_dataframe(self): try: from pandas import DataFrame except: raise Exception("pandas required to select dataframe. Please install" + "sudo easy_install pandas") columns = self.columns #check to see if they have duplicate column names if len(columns)>len(set(columns)): raise Exception("Cannot have duplicate column names " "in your query %s, please rename" % columns) return list_to_dataframe(self.data, columns)
def select_dataframe(self, query, args=()): """ Select everything into a datafrome with the column names being the names of the colums in the dataframe """ try: from pandas import DataFrame except: raise Exception("pandas required to select dataframe. Please install" + "sudo easy_install pandas") cursor = self.execute(query, args = args) data = cursor.fetchall() columns = [x[0].lower() for x in cursor.description] #check to see if they have duplicate column names if len(columns)>len(set(columns)): raise Exception("Cannot have duplicate column names " + "in your query %s, please rename" % columns) return list_to_dataframe(data, columns)