From 8b962d03eb866f3c75ef6d9ad0ac30dbebe5ebac Mon Sep 17 00:00:00 2001 From: Mattias Andrée Date: Thu, 6 Nov 2014 17:06:59 +0100 Subject: add support for non-numerical data MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Mattias Andrée --- src/median | 70 ++++++++++++++++++++++++++++++++++---------------------------- 1 file changed, 38 insertions(+), 32 deletions(-) diff --git a/src/median b/src/median index 34dcc0d..8684cef 100755 --- a/src/median +++ b/src/median @@ -36,43 +36,49 @@ lines.append((..., ...)) for value, key in lines: if last != key: if last is not None: - values.sort(key = lambda x : float(x)) + try: + values.sort(key = lambda x : float(x)) + except: + values.sort() if len(values) % 2 == 1: median = values[len(values) // 2] else: low = values[len(values) // 2 - 1] high = values[len(values) // 2 - 0] - low_plus, low_minus, low_dot = (low[0] == '+'), (low[0] == '-'), ('.' in low) - high_plus, high_minus, high_dot = (high[0] == '+'), (high[0] == '-'), ('.' in high) - if low_plus or low_minus: low = low[1:] - if high_plus or high_minus: high = high[1:] - if not low_dot: low += '.' - if not high_dot: high += '.' - lowi , lowd = [len(x) for x in low.split('.')] - highi, highd = [len(x) for x in high.split('.')] - if lowd < highd: low += '0' - if lowd > highd: high += '0' - mediani = max(lowi, highi) - mediand = max(lowd, highd) - median = float(low) * (-1 if low_minus else 1) - median += float(high) * (-1 if high_minus else 1) - negative, median = median < 0, abs(median) / 2 - if (int(low.replace('.', '')[-1]) + int(high.replace('.', '')[-1])) % 2 == 1: - mediand += 1 - if mediand == 0: - median = '%%0%ii' % mediani % median - if low_dot or high_dot: - median += '.' - else: - median = '%%0%i.%if' % (max(1, mediani) + mediand + 1, mediand) % median - if mediani == 0: - median = median[1:] - if low_minus and high_minus: prefix = '-' - elif negative: prefix = '-' - elif low_plus or high_plus: prefix = '+' - elif low_minus or high_minus: prefix = '0' - else: prefix = '' - median = prefix + median + try: + low_plus, low_minus, low_dot = (low[0] == '+'), (low[0] == '-'), ('.' in low) + high_plus, high_minus, high_dot = (high[0] == '+'), (high[0] == '-'), ('.' in high) + if low_plus or low_minus: low = low[1:] + if high_plus or high_minus: high = high[1:] + if not low_dot: low += '.' + if not high_dot: high += '.' + lowi , lowd = [len(x) for x in low.split('.')] + highi, highd = [len(x) for x in high.split('.')] + if lowd < highd: low += '0' + if lowd > highd: high += '0' + mediani = max(lowi, highi) + mediand = max(lowd, highd) + median = float(low) * (-1 if low_minus else 1) + median += float(high) * (-1 if high_minus else 1) + negative, median = median < 0, abs(median) / 2 + if (int(low.replace('.', '')[-1]) + int(high.replace('.', '')[-1])) % 2 == 1: + mediand += 1 + if mediand == 0: + median = '%%0%ii' % mediani % median + if low_dot or high_dot: + median += '.' + else: + median = '%%0%i.%if' % (max(1, mediani) + mediand + 1, mediand) % median + if mediani == 0: + median = median[1:] + if low_minus and high_minus: prefix = '-' + elif negative: prefix = '-' + elif low_plus or high_plus: prefix = '+' + elif low_minus or high_minus: prefix = '0' + else: prefix = '' + median = prefix + median + except: + median = values[len(values) // 2 - 1] print(median + last) last, values = key, [] values.append(value) -- cgit v1.2.3-70-g09d2