“大胃王”比赛数据柱形图绘制——柱形图展示优化

# -*- coding: utf-8 -*-

import pandas as pd

from matplotlib import pyplot as plt

hot_dog = pd.read_csv(r"matplotlib_bar/csv/hot-dog-contest-winners.csv")

def plot(): 

    # ********* Begin *********#

    fig,ax=plt.subplots()

    ax.bar(hot_dog["Year"],hot_dog["Dogs eaten"],width=[0.6],color=unitedStatesColor()) #添加指定的宽度  

    plt.show()

    ax.set_xlabel("Year")  #设置x轴标签  

    ax.set_ylabel("Dogs Eaten")  #设置y轴标签  

    ax.set_title("Hotdog game scores 1980-2010") #设置标题  

    ax.set_xlim(1979,2011)  #设置x轴数据限值  

    plt.rcParams['figure.figsize'] = (8.0, 4.0) #设置figure_size尺寸    

    plt.savefig('matplotlib_bar/studentfile/studentanswer/level_2/US.png')

    plt.close()

    plt.savefig('matplotlib_bar/studentfile/studentanswer/level_2/hotdog.png') #保存png文件  

    plt.close() #关闭画布窗口    

    # ********* End *********#

def unitedStatesColor(): 

    # ********* Begin *********#

    "打破记录的年份显示为粉红色，其余年份为灰绿色"  

    list=[]  

    for i in hot_dog["New record"]:  

        if 'country'=='United States':  

            list.append("#DB7093")  

        else:  

            list.append("#5F9F9F") 

    return list  

    # ********* End *********#