Add sparkline helper.
authorScott Gasch <[email protected]>
Sat, 5 Feb 2022 17:30:07 +0000 (09:30 -0800)
committerScott Gasch <[email protected]>
Sat, 5 Feb 2022 17:30:07 +0000 (09:30 -0800)
text_utils.py

index d9bb652689f8b317af606b4334a9e09604518975..76cc7e816db9ae87581989bc314fd7d07d4ca154 100644 (file)
@@ -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],
     *,