from pyecharts.charts import Bar,Timeline
from pyecharts.options import LabelOpts,TitleOpts
# 打开文件,注意编码格式。
with open("/Desktop/python/Project/数据可视化/1960-2019全球GDP数据.csv",'r',encoding="GB2312") as f:
# 逐行读取数据,返回的是一个列表。
f = f.readlines()
# 删除第一行没用元素
f.pop(0)
# 遍历列表中的元素
data_dic = {}
for x in f:
# 进行字符串分割,返回的是一个列表。
# 取出年份,城市和每个城市所对应的存放在一个字典中,格式为{年份:[[国家,gdp],[国家,gdp],[国家,gdp].........]}
# 对gdp进行处理使其转化为浮点数
data_list = x.split(",")
year = data_list[0]
city = data_list[1]
gdp = data_list[2]
gdp = float(gdp)
try:
# 直接追加列表可能会报KeyError,因为之前并没有这个关键字。
data_dic[year].append([city,gdp])
except KeyError:
# 报错的情况下把关键字的值创建成空列表,再往里面追加元素
data_dic[year] = []
data_dic[year].append([city, gdp])
# 排序年份,从字典中按照关键值进行排序,返回的是一个列表。
sorted_year = sorted(data_dic.keys())
# 定义排序函数的关键值,供sort使用。
def sort_choose(element):
return element[1]
# 创建时间线对
timeline = Timeline()
# 使用for循环遍历字典,取出每一年份中的数据,并对其使用sort进行从高到低的排序。
for year in sorted_year:
data_dic[year].sort(key=sort_choose,reverse=True)
# 使用列表切片取出列表中gdp最高的前八位。
top_8 = data_dic[year][:8]
# 构建X轴和Y轴空列表用来构建柱状图。
y_data = []
x_data = []
# 遍历前八个国家,将国家和gdp分别放进X轴列表和Y轴列表。
for temp in top_8:
x_data.append(temp[0])
y_data.append(temp[1])
# 构建柱状图,创建柱状图对象
bar = Bar()
# 反转数据,使gdp最高的国家在最上面显示。
x_data.reverse()
y_data.reverse()
# 设置每一年的标题
bar.set_global_opts(
title_opts=TitleOpts(
title=f" {year}年全球前-8-GDP"
)
)
# 添加X轴
bar.add_xaxis(x_data)
# 添加Y轴
bar.add_yaxis("GDP(亿)",y_data,label_opts=LabelOpts(position="right"))
# 反转X轴和Y轴
bar.reversal_axis()
# 时间线添加bar柱状图
timeline.add(bar,f"{year}年GDP")
# 设置自动播放等设置
timeline.add_schema(
# 自动播放的时间间隔,单位毫秒
play_interval=1000,
# 是否在自动播放的时候显示时间线
is_timeline_show=True,
# 是否自动播放
is_auto_play=True,
# 是否循环自动播放
is_loop_play=True
)
# 使用时间线绘图
timeline.render("1960-2019全球前八GDP.html")
效果演示: