Repository navigation
Expand file tree
/
Copy pathbarplot.py
More file actions
193 lines (165 loc) · 6.76 KB
/
Copy pathbarplot.py
File metadata and controls
193 lines (165 loc) · 6.76 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
# -*- coding: utf-8 -*-
"""
Plotting terminal based barplots
"""
from .utils.helpers import *
from .histogram import calc_bins, read_numbers
def plot_bar(f, height=20.0, bincount=None, binwidth=None, pch="o",
colour="default", title="", xlab=None, showSummary=False,
regular=False, return_str=False):
''' Plot bar.
Parameters
----------
f : list(number), numpy.ndarray, str(filepath)
input array
height : float
the height of the histogram in # of lines
bincount : int
number of bins in the histogram
binwidth : int
width of bins in the histogram
pch : str
shape of the bars in the plot, e.g 'o'
colour : str
white,aqua,pink,blue,yellow,green,red,grey,black,default,ENDC
title : str
title at the top of the plot, None = no title
xlab : boolean
whether or not to display x-axis labels
showSummary : boolean
whether or not to display a summary
regular : boolean
whether or not to start y-labels at 0
return_str : boolean
return string represent the plot or print it out, default: False
Example
-------
>>> y = np.random.rand(50)
>>> plot_bar(y, bincount=50, colour='red')
>>> 0.971|
>>> 0.923| o o o o
>>> 0.875| o o o o o oo
>>> 0.827| o o o o o o ooo
>>> 0.779| o o o o o o o o ooo
>>> 0.731| o o o o o o o o o ooo
>>> 0.683| oo oo o o o o o o o o o ooo
>>> 0.635| oo oo o oo o o o o o oo o ooo
>>> 0.587| oo oo o oo o ooo o o oo o ooo
>>> 0.539| oo oo o oooo ooo o o oo o ooo o
>>> 0.491| ooo ooo o oooo ooo o o ooo o ooo o
>>> 0.443| ooo ooo oo ooooo oooo o o ooo o ooo o
>>> 0.395| ooo ooo ooooooooo oooo o o ooo o ooo o
>>> 0.347| oooo ooo ooooooooo ooooo o o ooo o ooo o
>>> 0.299| oooo ooo ooooooooo ooooo o o ooo o ooo o
>>> 0.251| oooo ooo ooooooooo ooooo o o ooo o ooo o o
>>> 0.203| oooo ooo ooooooooo ooooo ooo ooo o ooo o o
>>> 0.155| oooo ooo oo ooooooooo ooooo ooo ooo ooooooo oo
>>> 0.107| ooooo ooo oo ooooooooo oooooooooo ooooooooooo ooo
>>> 0.059| ooooo oooooo ooooooooo ooooooooooooooooooooooo ooo
>>> 0.011| oooooooooooooooooooooo ooooooooooooooooooooooooooo
>>> --------------------------------------------------
'''
if pch is None:
pch = "o"
splot = ''
if isinstance(f, str):
f = open(f).readlines()
# ====== Create data ====== #
min_val, max_val = None, None
n, mean, sd = 0.0, 0.0, 0.0
# pick mode and get data
numbers = [i for i in read_numbers(f)]
int_mode = False
if numbers[0].is_integer():
int_mode = True
# rescale big enough to show on bars
min_orig = min(numbers) # original
max_orig = max(numbers)
numbers = [1000 * (i - min_orig) / (max_orig - min_orig + 1e-8) for i in numbers]
# statistics
n = len(numbers)
min_val = min(numbers)
max_val = max(numbers)
mean = sum(numbers) / n
sd = (sum([(mean - i)**2 for i in numbers]) / (n - 1)) ** 0.5
# bins is index
if bincount is not None:
bincount = min(bincount, n)
bins = list(calc_bins(n, 0., n + 0., bincount, binwidth))
bins = [int(i) for i in bins]
hist = dict((i, 0) for i in range(len(bins) - 1))
# hist is the mean value of array with indices within bin
for idx, (i, j) in enumerate(zip(bins, bins[1:])):
arr = numbers[i:j]
hist[idx] = sum(arr) / len(arr) # calculate mean
# ====== Start plot ====== #
min_y, max_y = min(hist.values()), max(hist.values())
start = max(min_y, 1)
stop = max_y + 1
if regular:
start = 1
if height is None:
height = stop - start
if height > 20:
height = 20
ys = list(drange(start, stop, float(stop - start) / height))
ys.reverse()
nlen = max(len(str(min_y)), len(str(max_y))) + 1
if title:
splot += print_return_str(
box_text(title, max(len(hist) * 2, len(title)), nlen),
return_str=return_str)
splot += print_return_str('', return_str=return_str)
used_labs = set()
for y in ys:
if int_mode:
ylab = '%d' % int(y * (max_orig - min_orig + 1e-8) / 1000 + min_orig)
else:
ylab = '%.3f' % float(y * (max_orig - min_orig + 1e-8) / 1000 + min_orig)
if ylab in used_labs:
ylab = ""
else:
used_labs.add(ylab)
ylab = " " * (nlen - len(ylab)) + ylab + "|"
splot += print_return_str(ylab, end=' ', return_str=return_str)
for i in range(len(hist)):
if int(y) <= hist[i]:
splot += printcolour(pch, True, colour, return_str)
else:
splot += printcolour(" ", True, colour, return_str)
splot += print_return_str('', return_str=return_str)
xs = hist.keys()
splot += print_return_str(" " * (nlen + 1) + "-" * len(xs),
return_str=return_str)
if xlab:
xlen = len(str(float((max_y) / height) + max_y))
for i in range(0, xlen):
splot += printcolour(" " * (nlen + 1), True, colour, return_str)
for x in range(0, len(hist)):
num = str(bins[x])
if x % 2 != 0:
pass
elif i < len(num):
splot += print_return_str(num[i], end=' ',
return_str=return_str)
else:
splot += print_return_str(" ", end=' ',
return_str=return_str)
splot += print_return_str('', return_str=return_str)
center = max(map(len, map(str, [n, min_val, mean, max_val])))
center += 15
if showSummary:
splot += print_return_str('', return_str=return_str)
splot += print_return_str("-" * (2 + center), return_str=return_str)
splot += print_return_str("|" + "Summary".center(center) + "|",
return_str=return_str)
splot += print_return_str("-" * (2 + center), return_str=return_str)
summary = "|" + ("observations: %d" % n).center(center) + "|\n"
summary += "|" + ("min value: %f" % min_val).center(center) + "|\n"
summary += "|" + ("mean : %f" % mean).center(center) + "|\n"
summary += "|" + ("sd : %f" % sd).center(center) + "|\n"
summary += "|" + ("max value: %f" % max_val).center(center) + "|\n"
summary += "-" * (2 + center)
splot += print_return_str(summary, return_str=return_str)
if return_str:
return splot