commit bce5f89882694abdab04addfbe663328ded113d9 Author: juga0 <juga@riseup.net> Date: Fri Aug 31 12:27:50 2018 +0000 Add method to obtain data for plotting --- sbws/lib/v3bwfile.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/sbws/lib/v3bwfile.py b/sbws/lib/v3bwfile.py index 65a99a8..04699c6 100644 --- a/sbws/lib/v3bwfile.py +++ b/sbws/lib/v3bwfile.py @@ -724,6 +724,15 @@ class V3BWFile(object): def median_bw(self): return median([l.bw for l in self.bw_lines]) + def to_plt(self, attrs=['bw'], sorted_by=None): + """Return bandwidth data in a format useful for matplotlib. + + Used from external tool to plot. + """ + x = [i for i in range(0, self.num)] + ys = [[getattr(l, k) for l in self.bw_lines] for k in attrs] + return x, ys, attrs + def write(self, output): if output == '/dev/stdout': log.info("Writing to stdout is not supported.")