在Python中如何绘制茎叶图

Question

在R中我们可以通过stem函数来绘制茎叶图,类似如下的图形:

那么在Python中有没有什么方式可以实现茎叶图的绘制?

Answer 1

group by就可以很方便的实现

from itertools import groupby
data = '89 79 57 46 1 24 71 5 6 9 10 15 16 19 22 31 40 41 52 55 60 61 65 69 70 75 85 91 92 94'

for k, g in groupby(sorted(data.split()), key=lambda x: int(x) // 10):
    lst = map(str, [int(_) % 10 for _ in list(g)])
    print k, '|', ' '.join(lst)