Seaborn regplot 与颜色条？

Question

我正在用 seaborn 的regplot 策划一些事情。据我了解，它在幕后使用pyplot.scatter。所以我假设如果我将散点图的颜色指定为一个序列，那么我就可以调用plt.colorbar，但它似乎不起作用：

sns.regplot('mapped both', 'unique; repeated at least once', wt, ci=95, logx=True, truncate=True, line_kws={"linewidth": 1, "color": "seagreen"}, scatter_kws={'c':wt['Cis/Trans'], 'cmap':'summer', 's':75})
plt.colorbar()

Traceback (most recent call last):

  File "<ipython-input-174-f2d61aff7c73>", line 2, in <module>
    plt.colorbar()

  File "/usr/local/lib/python2.7/dist-packages/matplotlib/pyplot.py", line 2152, in colorbar
    raise RuntimeError('No mappable was found to use for colorbar '

RuntimeError: No mappable was found to use for colorbar creation. First define a mappable such as an image (with imshow) or a contour set (with contourf).

为什么它不起作用，有没有办法解决它？

---

如果有一种简单的方法可以为尺寸生成图例，我会很好地使用点的大小而不是颜色

Answer 1

另一种方法是

import seaborn as sns
import matplotlib.pyplot as plt

tips = sns.load_dataset("tips")

points = plt.scatter(tips["total_bill"], tips["tip"],
                     c=tips["size"], s=75, cmap="BuGn")
plt.colorbar(points)

sns.regplot("total_bill", "tip", data=tips, scatter=False, color=".1")

Answer 2

regplot 的 color 参数将单一颜色应用于 regplot 元素（这在 seaborn 文档中）。要控制散点图，需要通过 kwargs 传递：

import pandas as pd
import seaborn as sns
import numpy.random as nr
import matplotlib.pyplot as plt

fig, ax = plt.subplots()

data = nr.random((9,3))
df = pd.DataFrame(data, columns=list('abc'))
out = sns.regplot('a','b',df, scatter=True,
                  ax=ax,
                  scatter_kws={'c':df['c'], 'cmap':'jet'})

然后您从 AxesSubplot seaborn 返回中获取可映射的事物（由 scatter 生成的集合），并指定您想要可映射的颜色条。请注意我的 TODO 评论，如果您打算对情节进行其他更改来运行它。

outpathc = out.get_children()[3] 
#TODO -- don't assume PathCollection is 4th; at least check type

plt.colorbar(mappable=outpathc)

plt.show()

Answer 3

事实上，您可以简单地访问 seaborn 绘图的图形对象并使用其colorbar() 函数手动添加颜色条。我发现这种方法要好得多。

此代码生成附加图：

import seaborn as sns
import matplotlib.pyplot as plt
tips = sns.load_dataset("tips")
sns.scatterplot(tips["total_bill"], tips["tip"],
                     hue=tips["size"], s=75, palette="BuGn", legend=False)
reg_plot=sns.regplot("total_bill", "tip", data=tips, scatter=False, color=".1")
reg_plot.figure.colorbar(mpl.cm.ScalarMappable([![enter image description here][1]][1]norm=mpl.colors.Normalize(vmin=0, vmax=tips["size"].max(), clip=False), cmap='BuGn'),label='tip size')