Использование оттенка 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_lo c (ключ) Файл "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'

Во время обработки вышеупомянутого исключения произошло другое исключение:

Traceback (последний вызов был последним): файл "C: / Users / krock / PycharmProjects / test / aaa.py", строка 9, в sns.pairplot (df.loc [:, 0: 6], hue = "label") Файл "C: \ Users \ krock \ PycharmProjects \ test \ venv \ lib \ site-packages \ seaborn \ axisgrid.py" , строка 2094, в паре высота = высота, аспект = аспект, dropna = dropna, ** grid_kws) Файл "C: \ Users \ krock \ PycharmProjects \ test \ venv \ lib \ site-packages \ seaborn \ axisgrid.py ", линия 1333, в init hue_names = utils.categorical_order (data [hue], hue_order) Файл "C: \ Users \ krock \ PycharmProjects \ test \ venv \ lib \ site-packages \ pandas \ core \ frame.py ", строка 2800, в getitem indexer = self.columns.get_lo c (ключ) Файл" C: \ Users \ krock \ PycharmProjects \ test \ venv \ lib \ site- packages \ pandas \ core \ indexes \ base.py ", строка 2648, в get_loc возвращает self._engine.get_lo c (self._maybe_cast_indexer (key)) Файл" pandas_libs \ index.pyx ", строка 111, в pandas ._ libs.index.IndexEngine.get_lo c Файл "pandas_libs \ index.pyx", строка 135, в pandas ._ libs.index.IndexEngine.get_lo c Файл "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 колонка.