数据可视化实验-茎叶图

数据可视化实验-茎叶图

根据1960年世界各国的出生率数据，控制台打印输出该年度各国出生率的茎叶图，茎叶图的茎为该年度各国出生率最小值与最大值之间，均匀间隔的整数，步长为2。

from itertools import groupby
import pandas as pd

# 加载数据
birth = pd.read_csv("birth-rate.csv")
# 删除空数据
birth.dropna(subset=['1960'], inplace=True)

dirt = {}
data = list(round(birth['1960'], 1))
range_num = []
for k, g in groupby(sorted(data), key=lambda x: int(x)):
    lst = map(str, list(map(lambda y: divmod(int(y * 10), 10)[1], list(g))))
    dirt[k] = ' '.join(lst)
    range_num.append(k)
num = list(range(range_num[0], range_num[-1], 2))
for i in num:
    a = ''
    for k in sorted(dirt.keys()):
        if 0 <= k - i <= 1:
            a = a + ' ' + dirt[k]
        elif k - i > 1:
            break
    print(str(i).rjust(5), '|', a)

实验结果