需求

使jointGrid畫圖的結(jié)果，在使用hue之后，仍然能保持一致。

給jointGrid添加title。

解決方案

我根據(jù)Stack Overflow上的一個(gè)方案做的修改。

def multivariateGrid(col_x, col_y, col_k, df, k_is_color=False, scatter_alpha=.5, title=""):def colored_scatter(x, y, c=None):def scatter(*args, **kwargs):args = (x, y)if c is not None:kwargs['c'] = ckwargs['alpha'] = scatter_alphaplt.scatter(*args, **kwargs)return scatterg = sns.JointGrid(x=col_x,y=col_y,data=df)g.fig.suptitle(title)g.fig.tight_layout()color = Nonelegends=[]for name, df_group in df.groupby(col_k):legends.append(name)if k_is_color:color=nameg.plot_joint(colored_scatter(df_group[col_x],df_group[col_y],color),)sns.distplot(df_group[col_x].values,ax=g.ax_marg_x,color=color,)sns.distplot(df_group[col_y].values,ax=g.ax_marg_y,color=color, vertical=True)# Do also global Hist:sns.distplot(df[col_x].values,ax=g.ax_marg_x,color='grey')sns.distplot(df[col_y].values.ravel(),ax=g.ax_marg_y,color='grey',vertical=True)plt.legend(legends)

使用方法跟，一般的jointplot一樣。只不過可以添加title并且保持顏色一致了。

multivariateGrid('x', 'y', 'kind', df=df, title="title test")

總結(jié)

以上是生活随笔為你收集整理的jointGrid，边际的颜色和hue保持一致，添加title的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。