Beeswarm Plot
Novel Length by Literary Genre
Book page count distribution across categories
Output
Python
import numpy as np
import matplotlib.pyplot as plt
np.random.seed(319)
BG_COLOR = '#ffffff'
TEXT_COLOR = '#1f2937'
COLORS = ['#F527B0', '#276CF5', '#F5B027', '#6CF527']
def simple_beeswarm(y, nbins=None, width=1.):
y = np.asarray(y)
if nbins is None:
nbins = np.ceil(len(y) / 6).astype(int)
nn, ybins = np.histogram(y, bins=nbins)
nmax = nn.max()
x = np.zeros(len(y))
ibs = []
for ymin, ymax in zip(ybins[:-1], ybins[1:]):
i = np.nonzero((y > ymin) * (y <= ymax))[0]
ibs.append(i)
dx = width / (nmax // 2) if nmax > 1 else width
for i in ibs:
yy = y[i]
if len(i) > 1:
j = len(i) % 2
i = i[np.argsort(yy)]
a = i[j::2]
b = i[j+1::2]
x[a] = (0.5 + j / 3 + np.arange(len(b))) * dx
x[b] = (0.5 + j / 3 + np.arange(len(b))) * -dx
return x
genres = ['Romance', 'Thriller', 'Fantasy', 'Biography']
data = {
'Romance': np.random.normal(280, 50, 50),
'Thriller': np.random.normal(320, 60, 55),
'Fantasy': np.random.normal(450, 100, 45),
'Biography': np.random.normal(350, 80, 48)
}
fig, ax = plt.subplots(figsize=(10, 6), facecolor=BG_COLOR)
ax.set_facecolor(BG_COLOR)
boxplot_data = []
for i, (genre, values) in enumerate(data.items()):
values = np.maximum(values, 100)
x = simple_beeswarm(values, width=0.3)
ax.scatter(x + i + 1, values, c=COLORS[i], alpha=0.7, s=45, edgecolors='white', linewidth=0.5)
boxplot_data.append(values)
bp = ax.boxplot(boxplot_data, positions=range(1, len(genres)+1), widths=0.5, patch_artist=True)
for patch in bp['boxes']:
patch.set_facecolor('none')
patch.set_edgecolor('#9ca3af')
for element in ['whiskers', 'caps', 'medians']:
for item in bp[element]:
item.set_color('#9ca3af')
ax.set_xticks(range(1, len(genres)+1))
ax.set_xticklabels(genres, color=TEXT_COLOR)
ax.set_xlabel('Genre', fontsize=12, color=TEXT_COLOR, fontweight='500')
ax.set_ylabel('Page Count', fontsize=12, color=TEXT_COLOR, fontweight='500')
ax.set_title('Novel Length by Genre', fontsize=14, color=TEXT_COLOR, fontweight='bold', pad=15)
ax.tick_params(colors='#374151', labelsize=10)
for spine in ax.spines.values():
spine.set_color('#e5e7eb')
plt.tight_layout()
plt.show()
Library
Matplotlib
Category
Statistical
More Beeswarm Plot examples
☕