Добавление легенды к коробчатому графику в matplotlib / seaborn - PullRequest
Новая
       87

Добавление легенды к коробчатому графику в matplotlib / seaborn

0 голосов
/ 28 мая 2020

Я новичок в Python.

Я создал коробчатую диаграмму (с наложением swarmplot) в matplotlib / seaborn, используя следующий код. Теперь я хотел бы добавить легенду, которая соответствует цветовой схеме каждого блока. Многие решения, которые я нахожу в Интернете, похоже, неприменимы к этому конкретному типу графика (например, работают только с сгруппированными коробчатыми диаграммами ).

Когда я пытаюсь реализовать код, предложенный здесь Я получаю сообщение об ошибке.

enter image description here

Весь ввод приветствуется! 

# Import libraries and modules

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns

# Set seaborn style.

sns.set(style="whitegrid", palette="colorblind")

# Load summary tidy data.

tidy = pd.read_csv('tidy.csv')

# Define plots for tidy data

fig, ax = plt.subplots(figsize=(10,6))
ax = sns.boxplot(x='header1', y='header2', data=tidy, order=["header1", "header2"])
ax = sns.swarmplot(x="header1", y="header2", data=tidy, color=".25", order=["header1", "header2"])
labels = [item.get_text() for item in ax.get_xticklabels()]
labels[0] = 'header1'
labels[1] = 'header2'
ax.set_xticklabels(labels)
ax.legend(loc='best')

Пример данных I работаю с. 

Object,Metric,Length
MT1,B2A1,3.57675
MT1,B2A2,2.9474600000000004
MT1,B2A3,2.247772857142857
MT1,B2A4,3.754455
MT1,B2A5,2.716282
MT1,B2A6,2.91325
MT10,B2A1,3.34361
MT10,B2A2,2.889958333333333
MT10,B2A3,2.22087
MT10,B2A4,2.87669
MT10,B2A5,1.6745005555555557
MT12,B2A1,3.3938900000000003
MT12,B2A2,2.00601
MT12,B2A3,2.1720200000000003
MT12,B2A4,2.452923333333333

1 Ответ

0 голосов
/ 02 июня 2020

Ошибка no handles with labels found to put in the legend вызвана вызовом ax.legend(), в то время как у ваших двух художников (boxplot и swarmplot) нет метки.

sns.boxplot основан на matplotlib boxplot и sns.swarmplot на scatter , поэтому все, что вам нужно, это дать им соответственно labels и label аргумент. 

ax = sns.boxplot(..., labels=["Metric", "Length"])
ax = sns.swarmplot(..., label="something goes here")

В качестве альтернативы, согласно this , вы можете оставить морскую часть нетронутой и поиграть с: 

handles, _ = ax.get_legend_handles_labels()          # Get the artists.
ax.legend(handles, ["label1", "label2"], loc="best") # Associate manually the artists to a label.
...