Визуализировать 3D-бар и график поверхности на той же фигуре в Python - PullRequest
0 голосов
/ 22 января 2019

Я хочу сравнить гистограмму двумерных случайных величин с ее распределением вероятностей.Я написал код для этого, и я получил оба графика на одной фигуре.Но я не могу различить два.Барный участок маскирует поверхностный участок.

Как я могу изменить цвета или любые другие параметры, чтобы я мог визуализировать их вместе и сравнить оба графика?

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

mu_x = 0
variance_x = 1
mu_y = 0
variance_y = 1
sample=1000

x, y = np.random.multivariate_normal((mu_x, mu_y), [[variance_x, 0], [0, variance_y]], sample).T
hist, xedges, yedges = np.histogram2d(x, y, bins=100, range=[[-10, 10], [-10, 10]],density=True)
xpos, ypos = np.meshgrid(xedges[:-1] + 0.25, yedges[:-1] + 0.25)
xpos = xpos.flatten('F')
ypos = ypos.flatten('F')
zpos = np.zeros_like(xpos)
dx = 0.5 * np.ones_like(zpos)
dy = dx.copy()
dz = hist.flatten()

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]])

fig = plt.figure()
ax = fig.gca(projection='3d')
ax.bar3d(xpos, ypos, zpos, dx, dy, dz, color='b', zsort='average')
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()

1 Ответ

0 голосов
/ 22 января 2019

Вы можете просто добавить смещение (вы можете указать любую сумму, которую вы предпочитаете) к гистограмме, так что оба упадут друг на друга, а затем наложить некоторую прозрачность на гистограмму, чтобы увидеть нижележащий контур.В подписи к рисунку можно упомянуть, что для целей представления добавлено смещение.В частности, вы можете просто использовать

ax.bar3d(xpos, ypos, zpos+0.35, dx, dy, dz, color='b', zsort='average')
ax.plot_surface(X, Y, rv.pdf(pos),cmap='viridis',linewidth=0)

, который производит

enter image description here

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...