用Python优雅地制作动态条形图
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:2] for 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这个库的功能，可以非常方便地制作条形图竞赛。
收工。😋
公众号“算法美食屋”后台回复关键字：动态图，可添加作者微信获取完整代码和人口数据集。
评论