diff --git a/plot2d.py b/plot2d.py index 0e90de4aaf076d8ef496fca498534a7d8996c699..f0e409be47ffc646064aa26730ed773adbe7bdfd 100644 --- a/plot2d.py +++ b/plot2d.py @@ -44,6 +44,13 @@ def scalarField(field, **kwargs): return res +def scalarFieldContour(field, **kwargs): + field = np.swapaxes(field, 0, 1) + res = contour(field, **kwargs) + axis('equal') + return res + + def multipleScalarFields(field, **kwargs): subPlots = field.shape[-1] for i in range(subPlots):