用Python优雅地制作动态条形图

共 2326字,需浏览 5分钟

 ·

2021-05-25 23:34

公众号“算法美食屋”后台回复关键字:动态图,可添加作者微信获取完整代码和人口数据集。


先上图片:


再上视频:



最后上代码:


import numpy as np 
import pandas as pd 
import pandas_alive
from matplotlib import pyplot as plt
import imageio

plt.rcParams['animation.writer'] = 'html'
plt.rcParams['font.family'] = ['sans-serif']
plt.rcParams['font.sans-serif'] = ['SimHei']

def html_to_gif(html_file, gif_file, duration=0.1):
    path = html_file.replace(".html","_frames")
    images = [os.path.join(path,x) for x in sorted(os.listdir(path))]
    frames = [imageio.imread(x) for x in images]
    imageio.mimsave(gif_file, frames, 'gif', duration=duration)
    return gif_file


df = pd.read_csv('./data/pop_amount.csv')
df["year"] = [str(x) for x in df["year"]]
df['year'] = pd.to_datetime(df['year'])
df = df.set_index("year")
df.columns = [x if "自治区" not in x else x[0:2for x in df.columns]

html_file = "population_bar_race.html"
gif_file = html_file.replace(".html",".gif")
mp4_file =  html_file.replace(".html",".mp4")

def current_total(values):
    total = values.sum()/10000
    s = '中国大陆总人口: {:.2f}亿'.format(total)
    return {'x'.90'y'.2's': s, 'ha''right''size'11}

chart = df.plot_animated(filename = html_file,title="中国大陆各省市历年人口变迁",
        figsize = (6,4),dpi=144,period_summary_func=current_total,
        n_visible=31,bar_texttemplate = '{x:.0f}',filter_column_colors = True,period_fmt="%Y年")

html_to_gif(html_file,gif_file,duration = 0.1)


主要是使用了pandas_alive这个库的功能,可以非常方便地制作条形图竞赛。


收工。😋


公众号“算法美食屋”后台回复关键字:动态图,可添加作者微信获取完整代码和人口数据集。


浏览 59
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报