searborn.pairplot() 中的 hue 参数不起作用

Question

各位。我正在尝试使用 searborn 绘制一对 IRIS 数据集图。为此，我按如下方式加载数据集：

import pandas as pd
iris = pd.read_csv('iris.csv')

数据集有四个数值特征和一个分类 class 称为多样性（有 3 个类别）。我使用以下简单代码对我的数据进行配对：

import searborn as sns
sns.pairplot(iris, hue = iris['variety'])

但是代码returns出现如下错误：

ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

有什么我遗漏的吗？我该如何解决这个问题？

Answer 1

参数hue只对应pandas里面的变量名。

因此，正确的代码是：

import seaborn as sns
sns.pairplot(iris, hue = 'variety')

如您所见，唯一的变化是我们提供 hue='variety' 而不是数组本身 (hue=iris['variety'])。

来源：

Seaborn 文档，pairplot 函数。

The hue parameter in searborn.pairplot() does not work