added visplot and matplotlib cross section

This commit is contained in:
Djalim Simaila 2023-04-21 17:54:22 +02:00
parent 533077d8b4
commit f220783aca
2 changed files with 32 additions and 0 deletions

View File

@ -8,4 +8,14 @@ def render2D(values:list):
fig = plt.figure()
ax = fig.add_subplot()
ax.plot(values)
plt.show()
def cross_section(x:list,y:list):
"""
Render a 2D model using matplotlib
:param values: A list with the values
"""
fig = plt.figure()
ax = fig.add_subplot()
ax.scatter(x,y)
plt.show()

View File

@ -0,0 +1,22 @@
import vispy.plot as vp
import numpy as np
def render2D(values:list):
fig = vp.Fig(size=(600, 500), show=False)
plotwidget = fig[0, 0]
fig.title = "bollu"
plotwidget.plot(values)
plotwidget.colorbar(position="top", cmap="autumn")
fig.show(run=True)
def cross_section(x:list,y:list):
color = (0.3, 0.5, 0.8)
fig = vp.Fig(show=False)
line = fig[0:4, 0:4].plot(np.column_stack((x,y)), symbol='o', width=0,
face_color=color + (0.02,), edge_color=None,
marker_size=8)
line.set_gl_state(depth_test=False)
fig.show(run=True)