matplotlib气泡图

import matplotlib.pyplot as plt

import pandas as pd

days=[]

animals=[]

level=[]

#color里放入对应的label项的颜色

color=['red','blue']

#label里放入对应的label的名字

label=['-','+']

#这里的sheet个数记得修改

sheet_size = 2

#excel路径在这里修改，最好把\改成/

#excel里三列要取名为day，animal，level

excel_path = 'D:/UserData/cyp/desktop/1.xlsx'

for i in range(1,sheet_size+1):

    df = pd.read_excel(excel_path,

                        sheet_name='Sheet'+str(i))

    days.append(df.loc[:,'day'])

    animals.append(df.loc[:,'animal'])

    level.append(df.loc[:,'level'])

for i in range(sheet_size):

    plt.scatter(days[i], animals[i], c=color[i], s=level[i]*10, label = label[i])

plt.xticks(range(1, 23, 1))

plt.yticks(range(1, 16, 1))

plt.grid(alpha=0.4)

plt.xlabel("Time(Day)", fontdict={'size': 12})

plt.ylabel("Animal_Numer", fontdict={'size': 12})

# 如果需要标题，把下面这行的#去掉，改里面的文字即可

# plt.title("历年天猫双11总成交额", fontdict={'size': 20})

plt.legend(loc='best')

plt.show()