python - 使用 seaborn，我如何在我的散点图中画一条我选择的线？

Question

我希望能够在 seaborn 中生成的绘图上画出我的规范线。我选择的图是 JointGrid，但任何散点图都可以。我怀疑 seaborn 可能不会让这件事变得容易？

这是绘制数据的代码(来自 Iris 数据集的花瓣长度和花瓣宽度的数据帧):

import seaborn as sns
iris = sns.load_dataset("iris")    
grid = sns.JointGrid(iris.petal_length, iris.petal_width, space=0, size=6, ratio=50)
    grid.plot_joint(plt.scatter, color="g")

如果您从 iris 数据集中获取此图，我如何在其上绘制一条我选择的线？例如，一条负斜率线可能将集群分开，而正斜率线可能会穿过它们。

Answer 1

看来您已将 matplotlib.pyplot 导入为 plt 以在您的代码中获取 plt.scatter。您可以只使用 matplotlib 函数来绘制直线:

import seaborn as sns
import matplotlib.pyplot as plt

iris = sns.load_dataset("iris")    
grid = sns.JointGrid(iris.petal_length, iris.petal_width, space=0, size=6, ratio=50)
grid.plot_joint(plt.scatter, color="g")
plt.plot([0, 4], [1.5, 0], linewidth=2)