update calc of figure size (inculde last commit:100c317), fixed #2

Author: jefferyUstc

dotplot/__init__.py

@@ -11,8 +11,8 @@
 
 class DotPlot(object):
     DEFAULT_ITEM_HEIGHT = 0.3
-    DEFAULT_ITEM_WIDTH = 0.35
-    DEFAULT_LEGENDS_WIDTH = .5
+    DEFAULT_ITEM_WIDTH = 0.3
+    DEFAULT_LEGENDS_WIDTH = .45
     MIN_FIGURE_HEIGHT = 3
 
     def __init__(self, df_size: pd.DataFrame,
@@ -42,11 +42,11 @@ def __get_figure(self):
         figure_width = mainplot_width + self.DEFAULT_LEGENDS_WIDTH
         plt.style.use('seaborn-white')
         fig = plt.figure(figsize=(figure_width, figure_height))
-        gs = gridspec.GridSpec(nrows=2, ncols=2, wspace=0.15, hspace=0.15,
+        gs = gridspec.GridSpec(nrows=3, ncols=2, wspace=0.15, hspace=0.15,
                                width_ratios=[mainplot_width, self.DEFAULT_LEGENDS_WIDTH])
         ax = fig.add_subplot(gs[:, 0])
-        ax_cbar = fig.add_subplot(gs[1, 1])
-        ax_legend = fig.add_subplot(gs[0, 1])
+        ax_cbar = fig.add_subplot(gs[2, 1])
+        ax_legend = fig.add_subplot(gs[0:2, 1])
         return ax, ax_cbar, ax_legend, fig
 
     @classmethod