我们经常看到的Bar Chart Race(柱形竞赛图),可以看到数据的呈现非常的直观。今天就一起来学习下如何生成和上面一样的柱形竞赛图。
1、导入Python库
import pandas as pd
import matplotlib.pyplot as plt
import matplotlib.ticker as ticker
import matplotlib.animation as animation
from IPython.display import HTML
2、加载数据集
这里使用的是城市人口数据集,加载我们想要的数据:其中,name为城市名称,group为城市所在区域。
df = pd.read_csv("data/city_populations.csv", usecols=['name', 'group', 'year', 'value'])
df.head()
3、初步处理数据
提取某一年的TOP10城市:
current_year = 2018
dff = df[df['year'].eq(current_year)].sort_values(by='value', ascending=True).head(10)
4、 绘制基础柱状图
fig, ax = plt.subplots(figsize=(15, 8))
ax.barh(dff['name'], dff['value'])
5、 调整样式(设置颜色、添加标签)重新绘制图片
'''
遇到问题没人解答?小编创建了一个Python学习交流QQ群991032883
寻找有志同道合的小伙伴,互帮互助,群里还有不错的视频学习教程和PDF电子书!
'''
colors = dict(zip(
["India", "Europe", "Asia", "Latin America", "Middle East", "North America", "Africa"],
["#adb0ff", "#ffb3ff", "#90d595", "#e48381", "#aafbff", "#f7bb5f", "#eafb50"]
))
group_lk = df.set_index(