mirror of https://github.com/E-Almqvist/hsf
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
22 lines
424 B
22 lines
424 B
3 years ago
|
#!/usr/bin/python3
|
||
|
|
import matplotlib.pyplot as plt
|
||
|
|
import numpy as np
|
||
|
|
from phys import *
|
||
|
|
|
||
|
|
x = np.arange(0, BOX_LENGTH, 0.00001)
|
||
|
|
n = np.arange(1, 10, 1)
|
||
|
|
|
||
|
|
X, N = np.meshgrid(x, n)
|
||
|
|
P = prob(X, N)
|
||
|
|
|
||
|
|
fig = plt.figure()
|
||
|
|
ax = plt.axes(projection='3d')
|
||
|
|
ax.plot_surface(X, N, P, rstride=1, cstride=1, cmap='viridis', edgecolor='none')
|
||
|
|
ax.set_xlabel('x [m]')
|
||
|
|
ax.set_ylabel('n [int]')
|
||
|
|
ax.set_zlabel('probability [frac]');
|
||
|
|
|
||
|
|
plt.show()
|
||
|
|
|
||
|
|
|