Matplotlib饼状图

fig = plt.figure()
count0 = data_train[data_train['Survived'] == 0].shape[0]
count1 = data_train[data_train['Survived'] == 1].shape[0]
labels = ['Survived=0','Survived=1']
sizes = [count0,count1]
colors = ['yellowgreen','lightskyblue']
plt.pie(sizes,labels = labels, colors=colors,autopct='%.2f%%',shadow=True)
plt.axis('equal') #保证饼状图是正圆，否则会出现倾斜
plt.show()