以下密度图与柱状图都是用Seaborn实现完成。
kedeplot实现密度图:
代码语言:javascript复制sns.set_style("whitegrid")
sns.kdeplot(train_data[train_data['Survived']==1]['Age'], bw=2, label="1", shade=True, color = 'red')
plt.xticks(np.arange(0, 90,5))
sns.kdeplot(train_data[train_data['Survived']==0]['Age'], bw=2, label="0", shade=True, color = 'blue' )
plt.xlabel('Age')
plt.ylabel('Density')
plt.title('Age')
plt.show()
sns.set_style: 设置主题,类似于R中的theme_set(). plt.xticks:修改x轴坐标。 plt.xlabel : 修改x轴名字。
distplot实现柱状图:
代码语言:javascript复制sns.distplot(merged_data_normal['Age'],kde=False, bins=20, hist = True,norm_hist=False, color = 'blue')
plt.title('Merged data')
plt.show()
或者用countplot实现:
代码语言:javascript复制var = 'Pclass'
sns.countplot(x=var, data=train_data, hue='Survived')
plt.title(var)
plt.legend(loc="upper right")
plt.show()
plt.title : 设置图的名字。 plt.legend(loc=) : 设置legend的位置。
countplot可以直接实现分组,方便快捷。