def test_to_dict_should_return_unicode_data_when_encoding_is_None(self):
my_table = Table(headers=["ham", "spam", "eggs"])
my_table.append({"ham": "á", "spam": "ç", "eggs": "à"})
my_dict = my_table.to_list_of_dicts(encoding=None)
self.assertEquals(len(my_dict), 1)
self.assertTrue(isinstance(my_dict[0]["ham"], unicode))
self.assertTrue(isinstance(my_dict[0]["spam"], unicode))
self.assertTrue(isinstance(my_dict[0]["eggs"], unicode))
self.assertEquals(my_dict[0]["ham"], u"á")
self.assertEquals(my_dict[0]["spam"], u"ç")
self.assertEquals(my_dict[0]["eggs"], u"à")
my_table = Table(headers=["ham", "spam", "eggs"], output_encoding="utf16")
my_table.append({"ham": "á", "spam": "ç", "eggs": "à"})
my_dict = my_table.to_list_of_dicts(encoding="iso-8859-1")
self.assertEquals(len(my_dict), 1)
self.assertEquals(my_dict[0]["ham"], u"á".encode("iso-8859-1"))
self.assertEquals(my_dict[0]["spam"], u"ç".encode("iso-8859-1"))
self.assertEquals(my_dict[0]["eggs"], u"à".encode("iso-8859-1"))
def test_to_dict_should_return_unicode_data_when_encoding_is_None(self):
my_table = Table(headers=['ham', 'spam', 'eggs'])
my_table.append({'ham': 'á', 'spam': 'ç', 'eggs': 'à'})
my_dict = my_table.to_list_of_dicts(encoding=None)
self.assertEquals(len(my_dict), 1)
self.assertTrue(isinstance(my_dict[0]['ham'], unicode))
self.assertTrue(isinstance(my_dict[0]['spam'], unicode))
self.assertTrue(isinstance(my_dict[0]['eggs'], unicode))
self.assertEquals(my_dict[0]['ham'], u'á')
self.assertEquals(my_dict[0]['spam'], u'ç')
self.assertEquals(my_dict[0]['eggs'], u'à')
my_table = Table(headers=['ham', 'spam', 'eggs'],
output_encoding='utf16')
my_table.append({'ham': 'á', 'spam': 'ç', 'eggs': 'à'})
my_dict = my_table.to_list_of_dicts(encoding='iso-8859-1')
self.assertEquals(len(my_dict), 1)
self.assertEquals(my_dict[0]['ham'], u'á'.encode('iso-8859-1'))
self.assertEquals(my_dict[0]['spam'], u'ç'.encode('iso-8859-1'))
self.assertEquals(my_dict[0]['eggs'], u'à'.encode('iso-8859-1'))
def test_list_of_dicts_should_handle_output_encoding_correctly(self):
my_table = Table(headers=["ham", "spam", "eggs"], output_encoding="utf16")
my_table.append((123, 456, 789))
my_table.append({"ham": "abc", "spam": "def", "eggs": "ghi"})
my_table.append((987, 654, 321))
my_dicts = my_table.to_list_of_dicts()
self.assertEquals(len(my_dicts), 3)
self.assertEquals(my_dicts[1][u"ham".encode("utf16")], u"abc".encode("utf16"))
self.assertEquals(my_dicts[1][u"spam".encode("utf16")], u"def".encode("utf16"))
self.assertEquals(my_dicts[1][u"eggs".encode("utf16")], u"ghi".encode("utf16"))
self.assertEquals(my_dicts[2][u"ham".encode("utf16")], 987)
self.assertEquals(my_dicts[2][u"spam".encode("utf16")], 654)
self.assertEquals(my_dicts[2][u"eggs".encode("utf16")], 321)
def test_to_dict_should_return_a_list_of_dict_with_headers_as_keys(self):
my_table = Table(headers=['ham', 'spam', 'eggs'])
my_table.append((123, 456, 789))
my_table.append({'ham': 'abc', 'spam': 'def', 'eggs': 'ghi'})
my_table.append((987, 654, 321))
my_dict = my_table.to_list_of_dicts()
self.assertEquals(len(my_dict), 3)
self.assertEquals(my_dict[0]['ham'], 123)
self.assertEquals(my_dict[0]['spam'], 456)
self.assertEquals(my_dict[0]['eggs'], 789)
self.assertEquals(my_dict[1]['ham'], 'abc')
self.assertEquals(my_dict[1]['spam'], 'def')
self.assertEquals(my_dict[1]['eggs'], 'ghi')
self.assertEquals(my_dict[2]['ham'], 987)
self.assertEquals(my_dict[2]['spam'], 654)
self.assertEquals(my_dict[2]['eggs'], 321)
def test_to_dict_should_return_a_list_of_dict_with_headers_as_keys(self):
my_table = Table(headers=["ham", "spam", "eggs"])
my_table.append((123, 456, 789))
my_table.append({"ham": "abc", "spam": "def", "eggs": "ghi"})
my_table.append((987, 654, 321))
my_dict = my_table.to_list_of_dicts()
self.assertEquals(len(my_dict), 3)
self.assertEquals(my_dict[0]["ham"], 123)
self.assertEquals(my_dict[0]["spam"], 456)
self.assertEquals(my_dict[0]["eggs"], 789)
self.assertEquals(my_dict[1]["ham"], "abc")
self.assertEquals(my_dict[1]["spam"], "def")
self.assertEquals(my_dict[1]["eggs"], "ghi")
self.assertEquals(my_dict[2]["ham"], 987)
self.assertEquals(my_dict[2]["spam"], 654)
self.assertEquals(my_dict[2]["eggs"], 321)
def test_list_of_dicts_should_handle_output_encoding_correctly(self):
my_table = Table(headers=['ham', 'spam', 'eggs'],
output_encoding='utf16')
my_table.append((123, 456, 789))
my_table.append({'ham': 'abc', 'spam': 'def', 'eggs': 'ghi'})
my_table.append((987, 654, 321))
my_dicts = my_table.to_list_of_dicts()
self.assertEquals(len(my_dicts), 3)
self.assertEquals(my_dicts[1][u'ham'.encode('utf16')],
u'abc'.encode('utf16'))
self.assertEquals(my_dicts[1][u'spam'.encode('utf16')],
u'def'.encode('utf16'))
self.assertEquals(my_dicts[1][u'eggs'.encode('utf16')],
u'ghi'.encode('utf16'))
self.assertEquals(my_dicts[2][u'ham'.encode('utf16')], 987)
self.assertEquals(my_dicts[2][u'spam'.encode('utf16')], 654)
self.assertEquals(my_dicts[2][u'eggs'.encode('utf16')], 321)
class Plotter(object):
'Stores information about a plot and plot it'
def __init__(self, data=None, rows=1, cols=1, width=1024, height=768):
self.rows = rows
self.cols = cols
self._subplot_number = 0
self.fig = figure(figsize=(width / 80, height / 80), dpi=80)
self._load_data(data)
def _load_data(self, data):
self.data = Table()
self.data.read('csv', data)
def _get_new_subplot(self, projection=None):
self._subplot_number += 1
if self._subplot_number > self.rows * self.cols:
raise OverflowError('This figure can handle only %d subplots' % \
self.rows * self.cols)
if projection is not None:
return self.fig.add_subplot(self.rows, self.cols,
self._subplot_number,
projection=projection)
else:
return self.fig.add_subplot(self.rows, self.cols,
self._subplot_number)
def save(self, filename):
#self.fig.savefig(filename, bbox_inches='tight', pad_inches=0.1)
self.fig.savefig(filename)
def linear(self, title='', grid=True, style='o-', x_labels=None,
legends=True, ignore='', colors=None,
colormap=matplotlib.cm.PRGn):
if legends is None or legends is True:
legends = {header: header for header in self.data.headers}
subplot = self._get_new_subplot()
subplot.set_title(title)
subplot.grid(grid)
columns_to_plot = []
for header in set(self.data.headers) - set(ignore):
if header != x_labels and self.data.types[header] in (int, float):
columns_to_plot.append(header)
if colors is None:
color_range = linspace(0, 0.9, len(columns_to_plot))
colors = [colormap(i) for i in color_range]
for header in columns_to_plot:
subplot.plot(self.data[header], style, label=legends[header],
color=colors.pop(0))
if x_labels is not None:
subplot.set_xticklabels(self.data[x_labels])
subplot.legend()
def scatter(self, x_column, title='', grid=True, labels=True, legends=True,
style='o-', ignore='', colors=None,
colormap=matplotlib.cm.PRGn, order_by=None, ordering='asc',
x_label=None, y_lim=None, legend_location='upper center',
legend_box=(0.5, 2.2), y_label=''):
subplot = self._get_new_subplot()
subplot.set_title(title)
subplot.grid(grid)
if order_by is not None:
self.data.order_by(order_by, ordering)
if legends is True:
legends = {header: header for header in self.data.headers}
if self.data.types[x_column] in (datetime.date, datetime.datetime):
self.fig.autofmt_xdate()
if labels:
if x_label is None:
x_label = x_column
subplot.set_xlabel(x_label)
subplot.set_ylabel(y_label)
x_values = range(1, len(self.data[x_column]) + 1)
subplot.set_xlim(0, max(x_values) + 1)
columns_to_plot = []
for header in set(self.data.headers) - set(ignore):
if header != x_column and self.data.types[header] in (int, float):
columns_to_plot.append(header)
if colors is None:
color_range = linspace(0, 0.9, len(columns_to_plot))
colors = [colormap(i) for i in color_range]
for header in columns_to_plot:
if legends is None:
subplot.plot(x_values, self.data[header], style,
color=colors.pop(0))
else:
subplot.plot(x_values, self.data[header], style,
label=legends[header], color=colors.pop(0))
subplot.set_xticks(x_values)
subplot.set_xticklabels(self.data[x_column])
if y_lim is not None:
subplot.set_ylim(y_lim)
if legends is not None:
subplot.legend(loc=legend_location, bbox_to_anchor=legend_box)
self.fig.subplots_adjust(top=0.5, right=0.9)
def bar(self, title='', grid=True, count=None, bar_width=0.8, x_column='',
bar_start=0.5, bar_increment=1.0, legends=True,
x_rotation=0, colors=None, colormap=matplotlib.cm.PRGn,
y_label=None, y_lim=None, y_columns=None):
if legends is True:
legends = {header: header for header in self.data.headers}
subplot = self._get_new_subplot()
subplot.set_title(title)
subplot.grid(grid)
bar_offset = (bar_increment - bar_width) / 2.0
bars_titles = []
if count is not None:
counter = Counter(self.data[count])
xticklabels = counter.keys()
columns_to_plot = [[counter[k] for k in xticklabels]]
if y_columns is not None:
columns_to_plot = y_columns
else:
columns_to_plot = []
if y_columns is None:
y_columns = self.data.headers[:]
for header in y_columns:
if self.data.types[header] in (int, float):
columns_to_plot.append(self.data[header])
bars_titles.append(header)
bar_width /= float(len(columns_to_plot))
bars = []
if colors is None:
color_range = linspace(0, 0.9, len(columns_to_plot))
colors = [colormap(i) for i in color_range]
for index, column in enumerate(columns_to_plot):
left = bar_start + index * bar_width
lefts = [bar_offset + left + i * bar_increment \
for i in range(len(column))]
bars.append(subplot.bar(lefts, column, bar_width,
color=colors.pop(0))[0])
xticks = [bar_start + bar_increment * (i + 0.5) \
for i in range(len(lefts))]
subplot.set_xticks(xticks)
subplot.set_xlim((min(xticks) - 2 * bar_start,
max(xticks) + 2 * bar_start))
if legends:
if count is None:
bars_titles = [legends[header] \
for header in bars_titles]
xticklabels = self.data[x_column]
else:
bars_titles = [legends[count]]
subplot.legend(bars, bars_titles)
else:
xticklabels = self.data[x_column]
subplot.set_xticklabels(xticklabels, rotation=x_rotation)
if y_label is not None:
subplot.set_ylabel(y_label)
if y_lim is not None:
subplot.set_ylim(y_lim)
def stacked_bar(self, x_column, y_column, y_labels=None, title='',
grid=True, bar_width=0.5, x_rotation=0, legends=True,
legend_location='upper left', legend_box=(-0.4, 1),
colors=None, colormap=matplotlib.cm.gist_heat):
subplot = self._get_new_subplot()
subplot.set_title(title)
subplot.grid(grid)
x_offset = (1.0 - bar_width) / 2
x_values_unique = list(set(self.data[x_column]))
x_values = array(range(len(set(self.data[x_column]))))
subplot.set_xticks(x_values + x_offset)
subplot.set_xticklabels(x_values_unique, rotation=x_rotation)
y_labels_values = list(set(self.data[y_labels]))
y_labels_values.sort()
data = {y: Counter() for y in y_labels_values}
if colors is None:
color_range = linspace(0, 0.9, len(data.keys()))
colors = [colormap(i) for i in color_range]
for row in self.data.to_list_of_dicts(encoding=None):
data[row[y_labels]][row[x_column]] += row[y_column]
bottom = zeros(len(x_values))
for y in y_labels_values:
values = [data[y][x] for x in x_values_unique]
subplot.bar(x_values, values, width=bar_width, label=unicode(y),
color=colors.pop(0), bottom=bottom)
bottom = [bottom[index] + value \
for index, value in enumerate(values)]
if legends:
subplot.legend(loc=legend_location, bbox_to_anchor=legend_box)
self.fig.subplots_adjust(bottom=0.1, left=0.25)
def radar(self, axis_labels, values, legends_column, title='',
x_grid=False, y_grid=True, fill_alpha=0.5, colors=None,
colormap=matplotlib.cm.gist_heat,
legend_location='upper left', legend_box=(-0.4, 1),
legends=False):
subplot = self._get_new_subplot(projection='polar')
subplot.set_title(title)
subplot.xaxis.grid(x_grid)
subplot.yaxis.grid(y_grid)
axis_labels_values = list(set(self.data[axis_labels]))
axis_labels_values.sort()
number_of_axis = len(axis_labels_values)
axis_angles = 2 * pi * linspace(0, 1 - 1.0 / number_of_axis,
number_of_axis)
subplot.set_thetagrids(axis_angles * 180 / pi, axis_labels_values)
legends_values = list(set(self.data[legends_column]))
legends_values.sort()
if colors is None:
len_legends = len(legends_values)
color_range = linspace(0, 1 - 1.0 / len_legends, len_legends)
colors = [colormap(i) for i in color_range]
curves = {x: Counter() for x in legends_values}
self.data.order_by(axis_labels)
for row in self.data.to_list_of_dicts(encoding=None):
curves[row[legends_column]][row[axis_labels]] += row[values]
for key in legends_values:
values = [curves[key][x] for x in axis_labels_values]
color = colors.pop(0)
lines = subplot.plot(axis_angles, values, color=color)
subplot.fill(axis_angles, values, facecolor=color,
alpha=fill_alpha)
x, y = lines[0].get_data()
new_x = concatenate((x, [x[0]]))
new_y = concatenate((y, [y[0]]))
lines[0].set_data(new_x, new_y)
if legends:
subplot.legend(legends_values, loc=legend_location,
bbox_to_anchor=legend_box)
def radar_area(self, values_column, labels_column, title='',
x_grid=False, y_grid=True, fill_alpha=0.5, colors=None,
colormap=matplotlib.cm.gist_heat, spacing=0.05):
subplot = self._get_new_subplot(projection='polar')
subplot.set_title(title)
subplot.xaxis.grid(x_grid)
subplot.yaxis.grid(y_grid)
values = self.data[values_column]
labels = self.data[labels_column]
if colors is None:
len_labels = len(labels)
color_range = linspace(0, 1 - 1.0 / len_labels, len_labels)
colors = [colormap(i) for i in color_range]
xticks = arange(0, 2 * pi, 2 * pi / len(labels)) + spacing / 2.0
width = xticks[1] - xticks[0] - spacing
subplot.bar(xticks, values, width=width, color=colors,
alpha=fill_alpha)
subplot.set_xticks(xticks + width / 2.0)
subplot.set_xticklabels(labels)
def pie(self, values_column, labels_column, title=''):
subplot = self._get_new_subplot()
subplot.pie(self.data[values_column], labels=self.data[labels_column],
autopct='%2.2f%%')
subplot.set_title(title)
#!/usr/bin/env python
# coding: utf-8
#A `Table` made with `dict`-like, `list`-like and `tuple`-like objects. For
#example, this code:
from outputty import Table
my_table = Table(headers=['First Name', 'Last Name', 'Main Language'])
my_table.append({'First Name': 'Álvaro', 'Last Name': 'Justen',
'Main Language': 'Python'})
my_table.append(('Flávio', 'Amieiro', 'Python'))
rows = my_table.to_list_of_dicts()
print rows[1]['First Name']
table_dict = my_table.to_dict()
print table_dict
table_dict_filtered = my_table.to_dict(only=['First Name', 'Last Name'])
print table_dict_filtered
other_table = Table(headers=['date', 'measure'])
other_table.append(('2011-12-01', 21))
other_table.append(('2011-12-02', 42))
other_table.append(('2011-12-03', 3.14))
other_table.append(('2011-12-04', 2.71))
values_as_dict = other_table.to_dict(key='date', value='measure')
print values_as_dict
class Plotter(object):
'Stores information about a plot and plot it'
def __init__(self, data=None, rows=1, cols=1, width=1024, height=768):
self.rows = rows
self.cols = cols
self._subplot_number = 0
self.fig = figure(figsize=(width / 80, height / 80), dpi=80)
self._load_data(data)
def _load_data(self, data):
self.data = Table()
self.data.read('csv', data)
def _get_new_subplot(self, projection=None):
self._subplot_number += 1
if self._subplot_number > self.rows * self.cols:
raise OverflowError('This figure can handle only %d subplots' % \
self.rows * self.cols)
if projection is not None:
return self.fig.add_subplot(self.rows,
self.cols,
self._subplot_number,
projection=projection)
else:
return self.fig.add_subplot(self.rows, self.cols,
self._subplot_number)
def save(self, filename):
#self.fig.savefig(filename, bbox_inches='tight', pad_inches=0.1)
self.fig.savefig(filename)
def linear(self,
title='',
grid=True,
style='o-',
x_labels=None,
legends=True,
ignore='',
colors=None,
colormap=matplotlib.cm.PRGn):
if legends is None or legends is True:
legends = {header: header for header in self.data.headers}
subplot = self._get_new_subplot()
subplot.set_title(title)
subplot.grid(grid)
columns_to_plot = []
for header in set(self.data.headers) - set(ignore):
if header != x_labels and self.data.types[header] in (int, float):
columns_to_plot.append(header)
if colors is None:
color_range = linspace(0, 0.9, len(columns_to_plot))
colors = [colormap(i) for i in color_range]
for header in columns_to_plot:
subplot.plot(self.data[header],
style,
label=legends[header],
color=colors.pop(0))
if x_labels is not None:
subplot.set_xticklabels(self.data[x_labels])
subplot.legend()
def scatter(self,
x_column,
title='',
grid=True,
labels=True,
legends=True,
style='o-',
ignore='',
colors=None,
colormap=matplotlib.cm.PRGn,
order_by=None,
ordering='asc',
x_label=None,
y_lim=None,
legend_location='upper center',
legend_box=(0.5, 2.2),
y_label=''):
subplot = self._get_new_subplot()
subplot.set_title(title)
subplot.grid(grid)
if order_by is not None:
self.data.order_by(order_by, ordering)
if legends is True:
legends = {header: header for header in self.data.headers}
if self.data.types[x_column] in (datetime.date, datetime.datetime):
self.fig.autofmt_xdate()
if labels:
if x_label is None:
x_label = x_column
subplot.set_xlabel(x_label)
subplot.set_ylabel(y_label)
x_values = range(1, len(self.data[x_column]) + 1)
subplot.set_xlim(0, max(x_values) + 1)
columns_to_plot = []
for header in set(self.data.headers) - set(ignore):
if header != x_column and self.data.types[header] in (int, float):
columns_to_plot.append(header)
if colors is None:
color_range = linspace(0, 0.9, len(columns_to_plot))
colors = [colormap(i) for i in color_range]
for header in columns_to_plot:
if legends is None:
subplot.plot(x_values,
self.data[header],
style,
color=colors.pop(0))
else:
subplot.plot(x_values,
self.data[header],
style,
label=legends[header],
color=colors.pop(0))
subplot.set_xticks(x_values)
subplot.set_xticklabels(self.data[x_column])
if y_lim is not None:
subplot.set_ylim(y_lim)
if legends is not None:
subplot.legend(loc=legend_location, bbox_to_anchor=legend_box)
self.fig.subplots_adjust(top=0.5, right=0.9)
def bar(self,
title='',
grid=True,
count=None,
bar_width=0.8,
x_column='',
bar_start=0.5,
bar_increment=1.0,
legends=True,
x_rotation=0,
colors=None,
colormap=matplotlib.cm.PRGn,
y_label=None,
y_lim=None,
y_columns=None):
if legends is True:
legends = {header: header for header in self.data.headers}
subplot = self._get_new_subplot()
subplot.set_title(title)
subplot.grid(grid)
bar_offset = (bar_increment - bar_width) / 2.0
bars_titles = []
if count is not None:
counter = Counter(self.data[count])
xticklabels = counter.keys()
columns_to_plot = [[counter[k] for k in xticklabels]]
if y_columns is not None:
columns_to_plot = y_columns
else:
columns_to_plot = []
if y_columns is None:
y_columns = self.data.headers[:]
for header in y_columns:
if self.data.types[header] in (int, float):
columns_to_plot.append(self.data[header])
bars_titles.append(header)
bar_width /= float(len(columns_to_plot))
bars = []
if colors is None:
color_range = linspace(0, 0.9, len(columns_to_plot))
colors = [colormap(i) for i in color_range]
for index, column in enumerate(columns_to_plot):
left = bar_start + index * bar_width
lefts = [bar_offset + left + i * bar_increment \
for i in range(len(column))]
bars.append(
subplot.bar(lefts, column, bar_width, color=colors.pop(0))[0])
xticks = [bar_start + bar_increment * (i + 0.5) \
for i in range(len(lefts))]
subplot.set_xticks(xticks)
subplot.set_xlim(
(min(xticks) - 2 * bar_start, max(xticks) + 2 * bar_start))
if legends:
if count is None:
bars_titles = [legends[header] \
for header in bars_titles]
xticklabels = self.data[x_column]
else:
bars_titles = [legends[count]]
subplot.legend(bars, bars_titles)
else:
xticklabels = self.data[x_column]
subplot.set_xticklabels(xticklabels, rotation=x_rotation)
if y_label is not None:
subplot.set_ylabel(y_label)
if y_lim is not None:
subplot.set_ylim(y_lim)
def stacked_bar(self,
x_column,
y_column,
y_labels=None,
title='',
grid=True,
bar_width=0.5,
x_rotation=0,
legends=True,
legend_location='upper left',
legend_box=(-0.4, 1),
colors=None,
colormap=matplotlib.cm.gist_heat):
subplot = self._get_new_subplot()
subplot.set_title(title)
subplot.grid(grid)
x_offset = (1.0 - bar_width) / 2
x_values_unique = list(set(self.data[x_column]))
x_values = array(range(len(set(self.data[x_column]))))
subplot.set_xticks(x_values + x_offset)
subplot.set_xticklabels(x_values_unique, rotation=x_rotation)
y_labels_values = list(set(self.data[y_labels]))
y_labels_values.sort()
data = {y: Counter() for y in y_labels_values}
if colors is None:
color_range = linspace(0, 0.9, len(data.keys()))
colors = [colormap(i) for i in color_range]
for row in self.data.to_list_of_dicts(encoding=None):
data[row[y_labels]][row[x_column]] += row[y_column]
bottom = zeros(len(x_values))
for y in y_labels_values:
values = [data[y][x] for x in x_values_unique]
subplot.bar(x_values,
values,
width=bar_width,
label=unicode(y),
color=colors.pop(0),
bottom=bottom)
bottom = [bottom[index] + value \
for index, value in enumerate(values)]
if legends:
subplot.legend(loc=legend_location, bbox_to_anchor=legend_box)
self.fig.subplots_adjust(bottom=0.1, left=0.25)
def radar(self,
axis_labels,
values,
legends_column,
title='',
x_grid=False,
y_grid=True,
fill_alpha=0.5,
colors=None,
colormap=matplotlib.cm.gist_heat,
legend_location='upper left',
legend_box=(-0.4, 1),
legends=False):
subplot = self._get_new_subplot(projection='polar')
subplot.set_title(title)
subplot.xaxis.grid(x_grid)
subplot.yaxis.grid(y_grid)
axis_labels_values = list(set(self.data[axis_labels]))
axis_labels_values.sort()
number_of_axis = len(axis_labels_values)
axis_angles = 2 * pi * linspace(0, 1 - 1.0 / number_of_axis,
number_of_axis)
subplot.set_thetagrids(axis_angles * 180 / pi, axis_labels_values)
legends_values = list(set(self.data[legends_column]))
legends_values.sort()
if colors is None:
len_legends = len(legends_values)
color_range = linspace(0, 1 - 1.0 / len_legends, len_legends)
colors = [colormap(i) for i in color_range]
curves = {x: Counter() for x in legends_values}
self.data.order_by(axis_labels)
for row in self.data.to_list_of_dicts(encoding=None):
curves[row[legends_column]][row[axis_labels]] += row[values]
for key in legends_values:
values = [curves[key][x] for x in axis_labels_values]
color = colors.pop(0)
lines = subplot.plot(axis_angles, values, color=color)
subplot.fill(axis_angles,
values,
facecolor=color,
alpha=fill_alpha)
x, y = lines[0].get_data()
new_x = concatenate((x, [x[0]]))
new_y = concatenate((y, [y[0]]))
lines[0].set_data(new_x, new_y)
if legends:
subplot.legend(legends_values,
loc=legend_location,
bbox_to_anchor=legend_box)
def radar_area(self,
values_column,
labels_column,
title='',
x_grid=False,
y_grid=True,
fill_alpha=0.5,
colors=None,
colormap=matplotlib.cm.gist_heat,
spacing=0.05):
subplot = self._get_new_subplot(projection='polar')
subplot.set_title(title)
subplot.xaxis.grid(x_grid)
subplot.yaxis.grid(y_grid)
values = self.data[values_column]
labels = self.data[labels_column]
if colors is None:
len_labels = len(labels)
color_range = linspace(0, 1 - 1.0 / len_labels, len_labels)
colors = [colormap(i) for i in color_range]
xticks = arange(0, 2 * pi, 2 * pi / len(labels)) + spacing / 2.0
width = xticks[1] - xticks[0] - spacing
subplot.bar(xticks,
values,
width=width,
color=colors,
alpha=fill_alpha)
subplot.set_xticks(xticks + width / 2.0)
subplot.set_xticklabels(labels)
def pie(self, values_column, labels_column, title=''):
subplot = self._get_new_subplot()
subplot.pie(self.data[values_column],
labels=self.data[labels_column],
autopct='%2.2f%%')
subplot.set_title(title)
#!/usr/bin/env python
# coding: utf-8
#A `Table` made with `dict`-like, `list`-like and `tuple`-like objects. For
#example, this code:
from outputty import Table
my_table = Table(headers=['First Name', 'Last Name', 'Main Language'])
my_table.append({
'First Name': 'Álvaro',
'Last Name': 'Justen',
'Main Language': 'Python'
})
my_table.append(('Flávio', 'Amieiro', 'Python'))
rows = my_table.to_list_of_dicts()
print rows[1]['First Name']
table_dict = my_table.to_dict()
print table_dict
table_dict_filtered = my_table.to_dict(only=['First Name', 'Last Name'])
print table_dict_filtered
other_table = Table(headers=['date', 'measure'])
other_table.append(('2011-12-01', 21))
other_table.append(('2011-12-02', 42))
other_table.append(('2011-12-03', 3.14))
other_table.append(('2011-12-04', 2.71))
values_as_dict = other_table.to_dict(key='date', value='measure')
print values_as_dict