在seaborn.jointplot中绘制两个分布

Dan*_*sen 8 python matplotlib pandas seaborn

pandas想在同一个seaborn jointplot中绘制两个数据帧.它看起来像这样(命令在IPython shell中; ipython --pylab):

import pandas as pd
import seaborn as sns
iris = sns.load_dataset('iris')
df = pd.read_csv('my_dataset.csv')
g = sns.jointplot('sepal_length', 'sepal_width', iris)
Run Code Online (Sandbox Code Playgroud)

两个数据帧中的键是相同的.
如何在同一个图中绘制我的值(当然是不同的颜色)?甚至更详细:如何绘制两个数据集,但只在顶部和侧面分配第一个?即只绘制点.

Jia*_* Li 24

这是通过修改基础数据来实现的一种方法sns.JointGrid.

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

# simulate some artificial data
# ========================================
np.random.seed(0)
data1 = np.random.multivariate_normal([0,0], [[1,0.5],[0.5,1]], size=200)
data2 = np.random.multivariate_normal([0,0], [[1,-0.8],[-0.8,1]], size=100)

# both df1 and df2 have bivaraite normals, df1.size=200, df2.size=100
df1 = pd.DataFrame(data1, columns=['x1', 'y1'])
df2 = pd.DataFrame(data2, columns=['x2', 'y2'])


# plot
# ========================================   
graph = sns.jointplot(x=df1.x1, y=df1.y1, color='r')

graph.x = df2.x2
graph.y = df2.y2
graph.plot_joint(plt.scatter, marker='x', c='b', s=50)
Run Code Online (Sandbox Code Playgroud)

在此输入图像描述

  • 如何在侧面的直方图中显示第二个分布? (12认同)

Jac*_*ari 5

在我看来,更好的解决方案是使用轴手柄来处理sns.joinplot返回的联合分布和边缘分布。使用这些(名称为ax_jointax_marg_xax_marg_y)也可以绘制边际分布图。

import seaborn as sns
import numpy as np

data1 = np.random.randn(100)
data2 = np.random.randn(100)
data3 = np.random.randn(100)
data4 = np.random.randn(100)

df1 = pd.DataFrame({'col1': data1, 'col2':data2})
df2 = pd.DataFrame({'col1': data3, 'col2':data4})

axs = sns.jointplot('col1', 'col2', data=df1)
axs.ax_joint.scatter('col1', 'col2', data=df2, c='r', marker='x')

# drawing pdf instead of histograms on the marginal axes
axs.ax_marg_x.cla()
axs.ax_marg_y.cla()
sns.distplot(df1.col1, ax=axs.ax_marg_x)
sns.distplot(df1.col2, ax=axs.ax_marg_y, vertical=True)
Run Code Online (Sandbox Code Playgroud)

边缘轴上的分布图