2017-08-09 49 views
-1

正如标题所提到的,我正在尝试创建4个matplotlib子图,并且在每个子图中我想绘制一个由我的数据框中的列调色的KDE图。我已经找到一种方法来“种”这样做,这是代码:Seaborn - 具有色调的KDE子图

ScoreDiffMinus1 = ScoreMovements.loc[ScoreMovements['ScoreDifference'] == -1] 
ScoreDiffMinus2 = ScoreMovements.loc[ScoreMovements['ScoreDifference'] == -2] 
ScoreDiffMinus3 = ScoreMovements.loc[ScoreMovements['ScoreDifference'] == -3] 
ScoreDiffMinus4 = ScoreMovements.loc[ScoreMovements['ScoreDifference'] == -4] 

fg, ax = plt.subplots(2,2, figsize = (20,12.5), sharex=True, sharey=True) 

Face1 = sns.FacetGrid(ScoreDiffMinus1, hue="Product") 
G1 = Face1.map(sns.kdeplot, 'DateDifference', ax = ax[0,0]) 

Face2 = sns.FacetGrid(ScoreDiffMinus2, hue="Product") 
G2 = Face2.map(sns.kdeplot, 'DateDifference', ax = ax[0,1]) 

Face3 = sns.FacetGrid(ScoreDiffMinus3, hue="Product") 
G3 = Face3.map(sns.kdeplot, 'DateDifference', ax = ax[1,0]) 

Face4 = sns.FacetGrid(ScoreDiffMinus4, hue="Product") 
G4 = Face4.map(sns.kdeplot, 'DateDifference', ax = ax[1,1]) 

这种方法的问题是,我同时运行一个FacetGrid和Matplotlib次要情节,这意味着我有4个图表和下面我有4个来自FacetGrid的空图表。这也意味着我不能像设置简单的KDE图一样设置xlabels,titles等。

有没有更好的方法来做到这一点?我觉得我错过了一些显而易见的事情,但上面的方法并不适合我。

感谢

+0

你需要选择你想要做的事情,道:你想使用matplotlib次要情节或者你要使用facetgrid。 – mwaskom

回答

0

怎么样以下

import matplotlib.pyplot as plt 
import pandas as pd 
import numpy as np 
import seaborn as sns 

p = np.random.choice(["Fan", "Table", "Ball"], size=100) 
sd = np.random.randint(-4,0,size=100) 
dd = np.random.randint(0,40,size=100) 
df = pd.DataFrame({'ScoreDifference': sd , 
        'DateDifference': dd , 
        "Product": p}) 



grid = sns.FacetGrid(df, col='ScoreDifference', hue="Product", col_order=range(-4,0)[::-1]) 
grid.map(sns.kdeplot, 'DateDifference') 
grid.add_legend() 

plt.show() 

enter image description here

+0

工作完美,非常感谢! – rubs90