Plot 3D mesh using 3D numpy array and 2D numpy array

Question

I have a 3D numpy array Lattice[][][], and a 2D numpy array Trajectory[][]. I want the indices i/j/k of Lattice to be the coordinate axes and the value contained in the tensor element to be plotted as a colored dot, making a 3D mesh maybe and I want to plot Trajectory where the Trajectory[p][0]/[1]/[2] contains the coordinates, and the index p is the iteration number - so this I want to plot as a 3D line, hopefully inside the mesh. I have no idea from where to start, is it possible in matplotlib?

Answer 1

You could definetly use matplotlib. Check this link:https://matplotlib.org/2.0.2/mpl_toolkits/mplot3d/tutorial.html

But you could also try a different package, like Plotly: https://plotly.com/python/3d-charts/