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

Question

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

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函数通过将color的值添加到绘图函数的**kwargs中而无意地管理了这些值。
如果要为散点图指定自己的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')