demo refactoring (#299)

Author: akharche

docs/demos/performance.py docs/demos/plotting.pydocs/demos/performance.py renamed to docs/demos/plotting.py

@@ -1,30 +1,28 @@
 import numpy as np
 import matplotlib.pyplot as plt
-# %matplotlib inline
 
 data_performance = {
 'read_parquet': ((0.804, 2.939, 6.235, 20.91), 
 	(0.51, 1.662, 3.151, 8.555)),
-'read_csv': ((5.526, 5.806, 7.008, 10.46), 
-	(19.732, 10.466,  6.277, 6.947)),
+'read_csv': ((19.732, 10.466, 6.277, 6.947)),
 'describe': ((0.823, 1.713, 3.324, 7.103), 
 	(0.704, 1.396, 2.89, 6.485)),
 'v_counts': ((0.867, 1.777, 3.447, 6.799), 
 	(0.699, 1.401, 2.914, 6.276)),
 }
 
 plot_params = {
-	'read_parquet':(('1m', '2m', '4m', '8m'), 'Data size'),
-	'read_csv':(('1 node', '2 nodes', '4 nodes', '8 nodes'), 'Number of processes'),
-	'describe':(('1m', '2m', '4m', '8m'), 'Data size'),
-	'v_counts':(('1m', '2m', '4m', '8m'), 'Data size'),
+	'read_parquet':(('1m', '2m', '4m', '8m'), 'Data size', 'Performance: Pandas vs SDC', True, 'upper left'),
+	'read_csv':(('1 node', '2 nodes', '4 nodes', '8 nodes'), 'Number of processes', 'SDC Scalability', False, 'upper right'),
+	'describe':(('1m', '2m', '4m', '8m'), 'Data size', 'Performance: Pandas vs SDC', True, 'upper left'),
+	'v_counts':(('1m', '2m', '4m', '8m'), 'Data size', 'Data size', 'Performance: Pandas vs SDC', True, 'upper left'),
 }
 
 
 class Plotter:
 	def __init__(self, func_id='read_parquet'):
 		self.func_id = func_id
-		self.x_labels, self.x_title = plot_params[self.func_id]
+		self.x_labels, self.x_title, self.title, self.is_compared, self.label_position = plot_params[self.func_id]
 		self.ngroups = len(self.x_labels)
 
 	def autolabel(self, rects, ax):
@@ -39,30 +37,37 @@ def autolabel(self, rects, ax):
 	def plot_performance(self):
 
 		plt.figure(figsize = (16, 8))
-
-		means_pandas, means_sdc = data_performance[self.func_id]
 		# create plot
 		index = np.arange(self.ngroups)
 		bar_width = 0.35
 		opacity = 0.8
 
-		rects1 = plt.bar(index, means_pandas, bar_width,
-		alpha=opacity,
-		label='Pandas')
+		plt.xlabel(self.x_title, fontsize=16)
+		plt.ylabel('Time, s', fontsize=16)
+		plt.title(self.title, fontsize=18)
+		
+		plt.tick_params(labelsize=12)
 
-		rects2 = plt.bar(index + bar_width, means_sdc, bar_width,
+		if self.is_compared:
+			data_pandas, data_sdc = data_performance[self.func_id]
+			rects_pandas = plt.bar(index + bar_width, data_pandas, bar_width,
+			alpha=opacity,
+			label='Pandas')
+
+			plt.xticks(index + bar_width, self.x_labels)
+		else:
+			data_sdc = data_performance[self.func_id]
+			plt.xticks(index, self.x_labels)
+
+		rects_sdc = plt.bar(index, data_sdc, bar_width,
 		alpha=opacity,
 		label='SDC')
 
-		plt.xlabel(self.x_title, fontsize=16)
-		plt.ylabel('Time, s', fontsize=16)
-		plt.title('Performance: Pandas vs SDC', fontsize=18)
-		plt.xticks(index + bar_width, self.x_labels)
-		plt.tick_params(labelsize=12)
-		plt.legend(fontsize=16,loc="upper left")
+		if self.is_compared:
+			self.autolabel(rects_pandas, plt)
 
-		self.autolabel(rects1, plt)
-		self.autolabel(rects2, plt)
+		plt.legend(fontsize=16, loc=self.label_position)
+		self.autolabel(rects_sdc, plt)
 
 		plt.tight_layout()
 		plt.show()