使用sns.pairplot色调

Question

df = pd.read_csv('model_predictions.csv', delimiter=',', header=None)
df = df.rename(columns={7: 'label'})
sns.set(style='ticks')
sns.pairplot(df, hue="label")
plt.show()

我使用这段代码生成一个配对图，并希望更改颜色以使用“label”列中的值，但下面会出现错误。

"C:\Users\krock\PycharmProjects\test\venv\lib\site-packages\pandas\core\indexes\base.py"，跟踪(最近一次调用)：文件

第2646行，get_loc返回self._engine.get_loc(键)文件"pandas_libs\index.pyx"，第111行，pandas._libs.index.IndexEngine.get_loc文件"pandas_libs\index.pyx"，第135行，在pandas._libs.index.IndexEngine.get_loc文件"pandas_libs\index_class_helper.pxi“中，第109行，pandas._libs.index.Int64Engine._check_type KeyError：'label‘

在处理上述异常的过程中，发生了另一个异常：

追溯(最近一次调用)：文件"C:/Users/krock/PycharmProjects/test/aaa.py"，第9行，在"C:\Users\krock\PycharmProjects\test\venv\lib\site-packages\seaborn\axisgrid.py"，sns.pairplot(df.loc: 0:6，hue="label")文件第2094行中，在对图height=height、aspect=aspect、dropna=dropna中，文件"C:\Users\krock\PycharmProjects\test\venv\lib\site-packages\seaborn\axisgrid.py"，第1333行，在init hue_names = utils.categorical_order(datahue，hue_order)文件"C:\Users\krock\PycharmProjects\test\venv\lib\site-packages\pandas\core\frame.py"，行2800中，在"C:\Users\krock\PycharmProjects\test\venv\lib\site-packages\pandas\core\indexes\base.py"，getitem indexer = self.columns.get_loc(key) 第2648行中，get_loc返回文件"pandas_libs\index.pyx"，第111行，pandas._libs.index.IndexEngine.get_loc文件"pandas_libs\index.pyx"，第135行，在pandas._libs.index.IndexEngine.get_loc文件"pandas_libs\index_class_helper.pxi“中，第109行，pandas._libs.index.Int64Engine._check_type KeyError：'label‘

进程已完成，退出代码为%1

我不知道怎么回事？

Answer 1

主要原因是python无法在子集label中找到列df.loc[:, 0:7]。

尝试使用完整的数据集，或者确保子集包含label列。