как добавить коробочный график для разброса данных в matplotlib

Question

У меня есть следующие данные, и после построения точечной точки разброса я хотел бы добавить коробку вокруг каждого набора позиций.Вот мой код для построения графика рассеяния:

%matplotlib inline
import matplotlib.pyplot as plt

X = [1, 1, 1, 1, 1, 1, 1, 
      2, 2, 2, 2, 2, 2, 2, 
     3, 3, 3, 3, 3, 3, 3,
     4, 4, 4, 4, 4, 4, 4,
     5, 5, 5, 5, 5, 5, 5,
     6, 6, 6, 6, 6, 6, 6,
     7, 7, 7, 7, 7, 7, 7,
     8, 8, 8, 8, 8, 8, 8,
     9, 9, 9, 9, 9, 9, 9,
     10, 10, 10, 10, 10, 10, 10,
     11, 11, 11, 11, 11, 11, 11,
     12, 12, 12, 12, 12, 12, 12,
     13, 13, 13, 13, 13, 13, 13,
     14, 14, 14, 14, 14, 14, 14,
     15, 15, 15, 15, 15, 15, 15]

H = [15, 17, 16, 20, 15, 18, 15,
      17, 16, 16, 20, 19, 18, 15,
      20, 22, 20, 22, 19, 21, 21,
      19, 21, 20, 23, 21, 20, 22,
      21, 23, 22, 20, 24, 22, 20,
      20, 19, 20, 18, 21, 17, 19,
      18, 20, 16, 15, 17, 20, 19,
       19, 19, 18, 21, 21, 16, 19,
       21, 22, 22, 24, 24, 23, 25,
       28, 26, 30, 27, 26, 29, 30,
       27, 26, 29, 31, 27, 29, 30,
       25, 26, 27, 28, 25, 27, 30,
      31, 28, 25, 27, 30, 25, 31,
      28, 26, 30, 28, 29, 27, 31,
      24, 26, 25, 28, 26, 23, 25]

fig, axes = plt.subplots(figsize=(8,5))
axes.scatter(X, H, color='b')
axes.set_xlabel('Pos');
axes.set_ylabel('H, µm');

, когда я добавляю plt.boxplot, он захватывает все данные, а не отдельные позиции.Я ценю ответы либо в Matplotlib или Seaborn.

спасибо

Answer 1

Хорошим способом было бы использование панд:

df = pd.DataFrame({'X':X, 'H': H})
ax=df.plot(kind='scatter', x='X', y='H')
df.boxplot(by='X', ax=ax)
plt.show()

Вывод:

Answer 2

Вот сокращенное решение о том, как сопоставить массив H с помощью X и построить его, используя matplotlib:

groups = [[] for i in range(max(X))]
[groups[X[i]-1].append(H[i]) for i in range(len(H))]
plt.boxplot(groups)

Результат:

Вы можете добавить сетку с помощью plt.grid(True)