使用 seaborn.facetgrid，如何指定映射散点图的颜色以反映数据框中列的值？

Question

我想创建散点图的 FacetGrid，其中点的颜色由绘制的数据框中的列定义。 但是，当我映射它时，似乎我无法将列名传递给plt.scatter的c=参数，因为它被解释为一串颜色而不是列名：

import seaborn as sns
import matplotlib.pyplot as plt

sns.set(style='white')

iris = sns.load_dataset('iris')
g = sns.FacetGrid(iris, row='species', size=4)
g.map(plt.scatter, 'sepal_width', 'sepal_length', c='petal_length')

出去：

/home/user/anaconda/lib/python2.7/site-packages/matplotlib/colors.pyc in to_rgba_array(self, c, alpha)
    420             result = np.zeros((nc, 4), dtype=np.float)
    421             for i, cc in enumerate(c):
--> 422                 result[i] = self.to_rgba(cc, alpha)
    423             return result
    424 

/home/user/anaconda/lib/python2.7/site-packages/matplotlib/colors.pyc in to_rgba(self, arg, alpha)
    374         except (TypeError, ValueError) as exc:
    375             raise ValueError(
--> 376                 'to_rgba: Invalid rgba arg "%s"\n%s' % (str(arg), exc))
    377 
    378     def to_rgba_array(self, c, alpha=None):

ValueError: to_rgba: Invalid rgba arg "p"
to_rgb: Invalid rgb arg "p"
could not convert string to float: p

我预期的结果与plt.scatter(iris.sepal_width, iris.sepal_length, c=iris.petal_length)

我简单地尝试了sns.regplot ，但它似乎遇到了同样的问题。 如果我不指定 FacetGrid 的row=或col=参数，我可以输入c=iris.petal_length以获得预期结果。

有没有办法创建一个 FacetGrid，其中数据按行或列分组，数据点根据数据框中的列着色？

Answer 1

在 DataFrame 中标识为列的变量需要与绘图函数中的位置参数相对应。 最简单的方法是在plt.scatter周围编写一个小包装函数，使其签名scatter(x, y, c)而不是scatter(x, y, s, c) ：

import seaborn as sns
import matplotlib.pyplot as plt

sns.set(style='white')

iris = sns.load_dataset('iris')
g = sns.FacetGrid(iris, row='species', size=4)

def scatter(x, y, c, **kwargs):
    plt.scatter(x, y, c=c, **kwargs)

g.map(scatter, 'sepal_width', 'sepal_length', 'petal_length')

Answer 2

这是你想要做的吗？

g.map(plt.scatter, 'sepal_width', 'sepal_length', c=iris.petal_length)

Answer 3

您可以通过指定hue参数来实现此目的。

g = sns.FacetGrid(iris, col='species', hue='petal_length', size=4)
g.map(plt.scatter, 'sepal_width', 'sepal_length')

这产生了这个情节： .

Answer 4

潜在的问题是 FacetGrid 的map函数通过将它们添加到绘图函数的**kwargs来无意中管理color值。
如果要为散点图指定自己的c或color ，可以在一个小包装器中从 kwargs 中删除 color 参数，然后指定您自己的参数，而不会出现任何错误：

def custom_scatter(x, y, c, **kwargs):
  del kwargs["color"]
  plt.scatter(x, y, c = c, **kwargs)

现在您可以使用custom_scatter进行映射：

g.map(custom_scatter, 'sepal_width', 'sepal_length', 'petal_length')