Python（八） pyecharts模块练习 1960年起 全球GDP柱状图

"""
扩展知识点
列表的排序sort
"""
from pyecharts.charts import Bar, Timeline
from pyecharts.options import LabelOpts

# 打开数据源文件，读取全部数据，关闭文件
f = open("E:\\BaiduNetdiskDownload\黑马Python\资料\可视化案例数据\动态柱状图数据\\1960-2019全球GDP数据.csv", "r",
         encoding="GB2312")
f_lines = f.readlines()  # readlines方法返回列表
f.close()
f_lines.pop(0)

# 将数据储存为字典 {年份：[国家, GDP]}
data_dict = {}

for v in f_lines:
    # replace 去除尾部的换行符，split分割字符串为数组
    list = v.replace("\n", "").split(",")

    year = int(list[0])  # 年份, 把年份转成int类型的数值
    country = list[1]  # 国家
    gdp = float(list[2])  # gdp, 有些gdp是用科学计数法的方式记录的，所有得转成浮点数

    # 因为字典里不存在的元素 直接调用的话灰报错，所以用try异常处理来判断字典里有没有 目标元素，如果没有的话就新建一个
    try:
        data_dict[year].append([country, gdp])
    except KeyError:
        data_dict[year] = []
        data_dict[year].append([country, gdp])

# 因为字典内部的元素是随机的 无序的，所以取出data_dict所有的key，并用sorted排序
sorted_year_list = sorted(data_dict.keys())

# 创建一个时间线对象
timeline = Timeline()

# 循环排序所有
for year in sorted_year_list:
    data_dict[year].sort(key=lambda element: element[1], reverse=True)

    # 取出前8名的国家
    year_dta = data_dict[year][0:8]

    # X轴添加国家 Y轴添加GDP
    x_data = []
    y_data = []
    for country_gdp in year_dta:
        x_data.append(country_gdp[0])  # 国家
        y_data.append(country_gdp[1] / 100000000)  # GDP

    # 创建柱状图, 添加数据, 反转XY轴
    bar = Bar()
    bar.add_xaxis(x_data)
    bar.add_yaxis("GDP(亿)", y_data, label_opts=LabelOpts(position="rigth"))
    bar.reversal_axis()

    # 添加时间轴数据
    timeline.add(bar, str(year))

# 自动播放设置
timeline.add_schema(
    play_interval=300,  # 播放速度
    is_timeline_show=True,  # 播放是否显示时间线
    is_auto_play=True,  # 是否自动播放
    is_loop_play=True,  # 是否循环播放
)
# 生成绘制文件
timeline.render("1960年 世界GDP时间柱状图.html")