def phase_2D(results,variable_x,variable_y,ax_xlim=None,ax_ylim=None): fig,ax = plt.subplots() #define axis limits if specified if(ax_xlim): ax.set_xlim(ax_xlim) if(ax_ylim): ax.set_ylim(ax_ylim) for result in results: ax.plot(result[:,functions.map_vars(variable_x)],result[:,functions.map_vars(variable_y)]) plt.show()
def phase_3D(results,variable_x,variable_y,variable_z,ax_xlim=None,ax_ylim=None,ax_zlim=None): fig = plt.figure() ax = fig.gca(projection='3d') #define axis limits if specified if(ax_xlim): ax.set_xlim(ax_xlim) if(ax_ylim): ax.set_ylim(ax_ylim) if(ax_zlim): ax.set_zlim(ax_zlim) for result in results: ax.plot(result[:,functions.map_vars(variable_x)],result[:,functions.map_vars(variable_y)],zs=result[:,functions.map_vars(variable_z)]) plt.show()