summaryrefslogtreecommitdiff
path: root/text_utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'text_utils.py')
-rw-r--r--text_utils.py25
1 files changed, 24 insertions, 1 deletions
diff --git a/text_utils.py b/text_utils.py
index d9bb652..76cc7e8 100644
--- a/text_utils.py
+++ b/text_utils.py
@@ -1,4 +1,5 @@
#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
"""Utilities for dealing with "text"."""
@@ -6,7 +7,7 @@ import logging
import math
import sys
from collections import defaultdict
-from typing import Dict, Generator, List, NamedTuple, Optional
+from typing import Dict, Generator, List, NamedTuple, Optional, Tuple
from ansi import fg, reset
@@ -105,6 +106,28 @@ def bar_graph(
)
+def sparkline(numbers: List[float]) -> Tuple[float, float, str]:
+ """
+ Makes a "sparkline" little inline histogram graph. Auto scales.
+
+ >>> sparkline([1, 2, 3, 5, 10, 3, 5, 7])
+ (1, 10, '▁▁▂▄█▂▄▆')
+
+ >>> sparkline([104, 99, 93, 96, 82, 77, 85, 73])
+ (73, 104, '█▇▆▆▃▂▄▁')
+
+ """
+ _bar = '▁▂▃▄▅▆▇█' # Unicode: 9601, 9602, 9603, 9604, 9605, 9606, 9607, 9608
+
+ barcount = len(_bar)
+ min_num, max_num = min(numbers), max(numbers)
+ span = max_num - min_num
+ sparkline = ''.join(
+ _bar[min([barcount - 1, int((n - min_num) / span * barcount)])] for n in numbers
+ )
+ return min_num, max_num, sparkline
+
+
def distribute_strings(
strings: List[str],
*,