aboutsummaryrefslogtreecommitdiffstatshomepage
diff options
context:
space:
mode:
authorEgor Tensin <Egor.Tensin@gmail.com>2017-01-27 02:34:44 +0300
committerEgor Tensin <Egor.Tensin@gmail.com>2017-01-27 02:34:44 +0300
commit3ee78cf60bc997c3e96926572f18f513fb99ae6f (patch)
tree7f6e474dfafe0a717d48f566377a280df3f289c9
parentrefactoring & bugfix (diff)
downloadvk-scripts-3ee78cf60bc997c3e96926572f18f513fb99ae6f.tar.gz
vk-scripts-3ee78cf60bc997c3e96926572f18f513fb99ae6f.zip
BarChartBuilder: bugfix
-rw-r--r--bin/utils/bar_chart.py29
1 files changed, 20 insertions, 9 deletions
diff --git a/bin/utils/bar_chart.py b/bin/utils/bar_chart.py
index 12b43b4..d916481 100644
--- a/bin/utils/bar_chart.py
+++ b/bin/utils/bar_chart.py
@@ -27,10 +27,20 @@ class BarChartBuilder:
return self._ax.get_xaxis()
def set_categories_axis_limits(self, start=None, end=None):
- self._ax.set_ylim(bottom=start, top=end)
+ bottom, top = self._ax.get_ylim()
+ if start is not None:
+ bottom = start
+ if end is not None:
+ top = end
+ self._ax.set_ylim(bottom=bottom, top=top)
def set_values_axis_limits(self, start=None, end=None):
- self._ax.set_xlim(left=start, right=end)
+ left, right = self._ax.get_xlim()
+ if start is not None:
+ left = start
+ if end is not None:
+ right = end
+ self._ax.set_xlim(left=left, right=right)
def enable_grid_for_categories(self):
self._get_categories_axis().grid()
@@ -103,6 +113,8 @@ class BarChartBuilder:
if min(values) >= 0:
self.set_values_axis_limits(start=0)
+ if np.isclose(max(values), 0.):
+ self.set_values_axis_limits(end=1)
elif max(values) < 0:
self.set_values_axis_limits(end=0)
@@ -119,15 +131,14 @@ if __name__ == '__main__':
import argparse
parser = argparse.ArgumentParser()
- parser.add_argument('--categories', nargs='*', default=[],
- help='categories')
- parser.add_argument('--values', nargs='*', type=float, default=[],
- help='values')
+ parser.add_argument('--categories', nargs='*', metavar='LABEL',
+ default=[])
+ parser.add_argument('--values', nargs='*', metavar='N',
+ default=[], type=float)
- parser.add_argument('--output', '-o',
- help='set output file path')
+ parser.add_argument('--output', '-o', help='set output file path')
- parser.add_argument('--middle', action='store_true',
+ parser.add_argument('--align-middle', action='store_true',
dest='labels_align_middle',
help='align labels to the middle of the bars')