Python, Matplotlib: Zeichnen von vertikalen Linien in 3D-Darstellung, wenn Daten unabhängig sind

Question

Ich habe eine zufällige walker in der (x, y) -Ebene und eine -log (bivariate Gaussian) in der (x, y, z) -Ebene. Diese beiden Datensätze sind im Wesentlichen unabhängig.

Ich möchte, sagen 5 (x, y) Paare der Random Walker und zeichnen vertikale Linien auf der Z-Achse und beenden Sie die vertikale Linie, wenn es die bivariate Gaussian "trifft".

Dies ist mein Code so weit:

import matplotlib as mpl 
import matplotlib.pyplot as plt 
import random 
import numpy as np 
import seaborn as sns 
import scipy 
from mpl_toolkits.mplot3d import Axes3D 
from matplotlib.mlab import bivariate_normal 

%matplotlib inline 

    # Data for random walk 

def randomwalk(): 

    mpl.rcParams['legend.fontsize'] = 10 

    xyz = [] 
    cur = [0, 0] 

    for _ in range(40): 
     axis = random.randrange(0, 2) 
     cur[axis] += random.choice([-1, 1]) 
     xyz.append(cur[:]) 

    # Get density 

    x, y = zip(*xyz) 
    data = np.vstack([x,y]) 
    kde = scipy.stats.gaussian_kde(data) 
    density = kde(data) 


    # Data for bivariate gaussian 

    a = np.linspace(-7.5, 7.5, 40) 
    b = a 
    X,Y = np.meshgrid(a, b) 
    Z = bivariate_normal(X, Y) 
    surprise_Z = -np.log(Z) 

    # Get random points from walker and plot up z-axis to the gaussian 

    M = data[:,np.random.choice(20,5)].T 

    # Plot figure 

    fig = plt.figure(figsize=(10, 7)) 

    ax = fig.gca(projection='3d') 
    ax.plot(x, y, 'grey', label='Random walk') # Walker 
    ax.scatter(x[-1], y[-1], c='k', marker='o') # End point 

    ax.legend() 

    surf = ax.plot_surface(X, Y, surprise_Z, rstride=1, cstride=1, 
     cmap = plt.cm.gist_heat_r, alpha=0.1, linewidth=0.1) 

    #fig.colorbar(surf, shrink=0.5, aspect=7, cmap=plt.cm.gray_r) 

    for i in range(5): 
     ax.plot([M[i,0], M[i,0]],[M[i,1], M[i,1]], [0,10],'k--',alpha=0.8, linewidth=0.5) 

    ax.set_zlim(0, 50) 
    ax.set_xlim(-10, 10) 
    ax.set_ylim(-10, 10) 

Welche

Wie Sie das einzige, was mit ich bin zu kämpfen sehen erzeugt ist, wie die vertikalen Linien zu beenden, wenn sie erfüllen Sie den entsprechenden Z-Wert. Irgendwelche Ideen sind willkommen!

Answer 1

Sie können diese Zeilen derzeit nur auf eine Höhe von 10 setzen, indem Sie [0,10] als z-Koordinaten verwenden. Sie können Ihre Schleife wie folgt ändern:

for i in range(5): 
    x = [M[i,0], M[i,0]] 
    y = [M[i,1], M[i,1]] 
    z = [0,-np.log(bivariate_normal(M[i,0],M[i,1]))] 
    ax.plot(x,y,z,'k--',alpha=0.8, linewidth=0.5) 

Dies nimmt die x und y für jeden Punkt, den Sie Schleife über Koordinaten und berechnet die Höhe des darüber liegenden Gaußschen für diesen Punkt und Plots dort. Hier ist ein Diagramm mit dem Linienstil geändert, um die Linien für die Frage zu betonen: