matplotlib.mlab.bivariate_normal 的等效函數

Question

從 Matplotlib 3.1 開始，mlab.bivariate_normal 已從 matplotlib 的庫中刪除。 我想知道是否有一個內置類可以在其他地方（或在 matplotlib 中）做同樣的工作？ 我搜索了一下，但無法輕易找到。

我意識到可以在這里復制函數並使用它，但我想知道是否有一個可以調用的內置函數。

Answer 1

正如@tmdavison 所建議的， scipy.stats.multivariate_normal()可用於替換matplotlib.mlab.bivariate_normal() 。

電話：

Z = matplotlib.mlab.bivariate_normal(X, Y, sig_x, sig_y, mu_x, mu_y, sig_xy)

變成：

rv = scipy.stats.multivariate_normal([mu_x, mu_y], [[sig_x, sig_xy], [sig_xy, sig_y]])
Z = rv.pdf(np.dstack((X, Y)))

matplotlib.mlab.bivariate_normal()函數對除X和Y之外的所有參數都有默認值，因此您可能需要添加一些額外的值。 sig_x和sig_y值默認為1.0 ，所有其他參數默認為0.0 。 因此，如果您要替換調用bivariate_normal(X, Y)它將變為multivariate_normal([0, 0], [[1, 0], [0, 1]]).pdf(np.dstack((X, Y))) 。