@@ -141,18 +141,6 @@ def sgrid(scaling=None):
141
141
return ax ,fig
142
142
143
143
144
- # Utility function used by all grid code
145
- def _final_setup (ax ,scaling = None ):
146
- ax .set_xlabel ('Real' )
147
- ax .set_ylabel ('Imaginary' )
148
- ax .axhline (y = 0 ,color = 'black' ,lw = 0.25 )
149
- ax .axvline (x = 0 ,color = 'black' ,lw = 0.25 )
150
-
151
- # Set up the scaling for the axes
152
- scaling = 'equal' if scaling is None else scaling
153
- plt .axis (scaling )
154
-
155
-
156
144
# If not grid is given, at least separate stable/unstable regions
157
145
def nogrid (dt = None ,ax = None ,scaling = None ):
158
146
fig = plt .gcf ()
@@ -226,3 +214,15 @@ def zgrid(zetas=None, wns=None, ax=None, scaling=None):
226
214
227
215
_final_setup (ax ,scaling = scaling )
228
216
return ax ,fig
217
+
218
+
219
+ # Utility function used by all grid code
220
+ def _final_setup (ax ,scaling = None ):
221
+ ax .set_xlabel ('Real' )
222
+ ax .set_ylabel ('Imaginary' )
223
+ ax .axhline (y = 0 ,color = 'black' ,lw = 0.25 )
224
+ ax .axvline (x = 0 ,color = 'black' ,lw = 0.25 )
225
+
226
+ # Set up the scaling for the axes
227
+ scaling = 'equal' if scaling is None else scaling
228
+ plt .axis (scaling )