Matplotlib_2 对应练习题

问题1 提出

• 使用提供的drug数据集，对第一列yyyy和第二列state分组求和，画出下面折线图。PA加粗标黄，其他为灰色。
  图标题和横纵坐标轴标题，以及线的文本暂不做要求。

问题1 解答

import matplotlib.pyplot as plt
import pandas as pd
import numpy as np

df = pd.read_csv('C:/Users/86137/Desktop/python note/学习计划2/fantastic-matplotlib-main/fantastic-matplotlib-main/data/Drugs.csv')
print(df.head())
df_new=df.groupby(['YYYY','State'])['DrugReports'].sum() # 分组求和
print(df_new.head(10))
x=np.arange(2010,2018,1)
fig, ax = plt.subplots()
list_1=['KY','OH','PA','VA','WV']
a=0
for i in list_1:
    if a==2:
        ax.plot(x, df_new[a::5], label=i, color='y',linewidth=5)
    else:
        ax.plot(x, df_new[a::5], label=i, color='grey')
    a+=1
ax.set_title(label="Evolution of PA vs other states", loc='left', color='orange')
ax.set_ylabel('DrugReports')
plt.grid()
plt.annotate('PA',(2017,df_new[-3::4]))
plt.show()

运行结果：

问题2 提出

• 分别用一组长方形柱和填充面积的方式模仿画出下图，函数 y = -1 * (x - 2) * (x - 8) +10 在区间[2,9]的积分面积
  
  

问题2 解答

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0, 10, 100)
y = -1 * (x -2) * (x - 8) + 10
fig, (ax1, ax2) = plt.subplots(2, 1)

ax1.set_ylim(0, 20)
ax1.set_xticks(range(0, 12, 2))
ax1.plot(x, y, color='red')
ax1.bar(x[20:90], y[20:90], alpha=0.5, width=0.06, color='grey', lw=0.05)

ax2.set_ylim(0, 20)
ax2.set_xticks(range(0, 12, 2))
ax2.plot(x, y, color='red')
ax2.fill_between(x[20:90],y[20:90],color='grey', alpha=0.3)

plt.show()

运行结果