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 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)
