python中的密度图与柱状图

2020-09-07 09:58:42 浏览数 (3)

以下密度图与柱状图都是用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可以直接实现分组,方便快捷。

0 人点赞