在3D中绘制正态分布

问题描述

我正在尝试绘制两个正态分布变量的公共分布.

I am trying to plot the comun distribution of two normal distributed variables.

下面的代码绘制了一个正态分布变量.绘制两个正态分布变量的代码是什么?

The code below plots one normal distributed variable. What would the code be for plotting two normal distributed variables?

import matplotlib.pyplot as plt
import numpy as np
import matplotlib.mlab as mlab
import math

mu = 0
variance = 1
sigma = math.sqrt(variance)
x = np.linspace(-3, 3, 100)
plt.plot(x,mlab.normpdf(x, mu, sigma))

plt.show()

推荐答案

听起来您正在寻找的是多元正态分布.这在scipy中以 scipy实现. stats.multivariate_normal .重要的是要记住，您正在将协方差矩阵传递给函数.因此，为了简单起见，请将对角线元素设为零:

It sounds like what you're looking for is a Multivariate Normal Distribution. This is implemented in scipy as scipy.stats.multivariate_normal. It's important to remember that you are passing a covariance matrix to the function. So to keep things simple keep the off diagonal elements as zero:

[X variance ,     0    ]
[     0     ,Y Variance]

这里是使用此功能并生成结果分布的3D图的示例.我添加了颜色图，以使查看曲线更容易，但可以随时将其删除.

Here is an example using this function and generating a 3D plot of the resulting distribution. I add the colormap to make seeing the curves easier but feel free to remove it.

import numpy as np
import matplotlib.pyplot as plt
from scipy.stats import multivariate_normal
from mpl_toolkits.mplot3d import Axes3D

#Parameters to set
mu_x = 0
variance_x = 3

mu_y = 0
variance_y = 15

#Create grid and multivariate normal
x = np.linspace(-10,10,500)
y = np.linspace(-10,10,500)
X, Y = np.meshgrid(x,y)
pos = np.empty(X.shape + (2,))
pos[:, :, 0] = X; pos[:, :, 1] = Y
rv = multivariate_normal([mu_x, mu_y], [[variance_x, 0], [0, variance_y]])

#Make a 3D plot
fig = plt.figure()
ax = fig.gca(projection='3d')
ax.plot_surface(X, Y, rv.pdf(pos),cmap='viridis',linewidth=0)
ax.set_xlabel('X axis')
ax.set_ylabel('Y axis')
ax.set_zlabel('Z axis')
plt.show()

为您提供此情节:

Giving:

这篇关于在3D中绘制正态分布的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！