python - 如何使用 Matplotlib 制作具有光环效果的散点图？

Question

我想绘制高斯随机变量的潜在空间其中 x_i~Gaussian(u_i, simga_i)我想像这张图一样代表情节中的不确定性 但是我找不到任何关于如何使用 Matplotlib 执行此操作的 API!

Answer 1

或许您可以试试下面的例子？ (修改自 here )

import numpy as np
import matplotlib.pyplot as plt

def normal_pdf(x, mean, var):
    return np.exp(-(x - mean)**2 / (2*var))

xmin, xmax, ymin, ymax = (0, 100, 0, 100)

N = 1000
xx = np.linspace(xmin, xmax, N)
yy = np.linspace(ymin, ymax, N)

x = normal_pdf(xx, 50, 75)
y = normal_pdf(yy, 50, 75)

weights = np.array(np.meshgrid(x, y)).prod(0)

plt.imshow(weights, extent=(xmin, xmax, ymin, ymax), cmap=plt.cm.Blues, alpha=0.5)
plt.plot(50, 50, "bo")
plt.tight_layout()
plt.show()

结果: