matplotlib画图9

import pandas as pd
import matplotlib.pyplot as plt
import matplotlib as mpl
import seaborn as sns
import numpy as np
plt.rcParams['font.sans-serif']=['SimHei']


#年
year=[1949,1949,1949,1949,1949,1949,
      1950,1950,1950,1950,1950,1950,
      1951,1951,1951,1951,1951,1951,
      1952,1952,1952,1952,1952,1952,]
#月
month=['jan','feb','mar','apr','may','jun',
       'jan','feb','mar','apr','may','jun',
       'jan','feb','mar','apr','may','jun',
       'jan','feb','mar','apr','may','jun',]
#
pas=[112,118,132,128,121,115,
     112,145,132,134,121,115,
     112,118,132,128,121,115,
     112,118,132,154,133,115,]

#将列表数据变为dataframe,在变csv,dataframe

df=pd.DataFrame({"year":year,"month":month,"pas":pas})
df.to_csv('passengers.csv')
df2=pd.read_csv("passengers.csv")
#按年份分组
df3=df2.groupby('year').sum()
print(df3)
#画图
plt.subplot(1,2,1)
plt.plot(df3['pas'],marker='o')
plt.xlabel('年份')
plt.ylabel('乘客')
plt.title("年份——乘客曲线图")
#按月份
df4=df2.groupby('month').sum()
plt.subplot(1,2,2)
plt.bar(df2.loc[0:5,'month'],df4['pas'])
plt.xlabel('yue份')
plt.ylabel('乘客')
plt.title("yue份——乘客曲线图")
plt.show()