Matplotlib可视化

Matplotlib图形配置
第1关 配置颜色条
import matplotlib
matplotlib.use("Agg")
import matplotlib.pyplot as plt
import numpy as np




def student(data):
    # ********* Begin *********#
    plt.figure(figsize=(10,10))
    plt.imshow(data, cmap='hot')
    plt.colorbar()
    plt.savefig("Task1/img/T1.png")
    plt.show()






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


第2关 设置注释
import matplotlib
matplotlib.use('Agg')
import pandas as pd
import matplotlib.pyplot as plt
import numpy as np
import warnings
warnings.filterwarnings("ignore")
import matplotlib as mpl

def student(file_name):
    plt.figure(figsize=(10,10))
    a=pd.read_csv(file_name).dropna()
    a=a["births"].groupby(a["year"]).sum()
    plt.plot(a)
    plt.annotate("max" ,(np.argmax(a) ,a.max()),(np.argmax(a)-5,a.max(