在 python、matplotlib、多臂 bandit 中绘制正态分布

Question

如何在 matplotlib 中绘制多个类似正态分布的图像，如上面的图像？

Answer 1

这种情节被称为“小提琴”情节：

import numpy as np
from matplotlib import pyplot as plt

# Generate 5 actions with different reward distributions
data = np.random.randn(1000, 5) * [1, 0.5, 1.5, 1, 0.5] + [0, 1, -1, 2, -0.5]

plt.violinplot(data)
plt.xlabel('Action')
plt.ylabel('Reward distribution')