|
187 | 187 | "my_cmap = ListedColormap(my_cmap)" |
188 | 188 | ] |
189 | 189 | }, |
190 | | - { |
191 | | - "cell_type": "code", |
192 | | - "execution_count": null, |
193 | | - "metadata": { |
194 | | - "hide_input": true |
195 | | - }, |
196 | | - "outputs": [], |
197 | | - "source": [ |
198 | | - "fig, (ax1, ax2, ax3) = plt.subplots(1, 3, figsize=(15, 5), constrained_layout=True)\n", |
199 | | - "fig.suptitle(\"Figure 1\", fontsize=16)\n", |
200 | | - "\n", |
201 | | - "ax1.set_xlabel(\"Longitude\")\n", |
202 | | - "ax1.set_ylabel(\"Latitude\")\n", |
203 | | - "ax1.set_title(\n", |
204 | | - " \"A) Node at corner cell\\nAgreement with Parcels interpolation\",\n", |
205 | | - " fontsize=14,\n", |
206 | | - " fontweight=\"bold\",\n", |
207 | | - ")\n", |
208 | | - "ax1.pcolormesh(\n", |
209 | | - " x_centers,\n", |
210 | | - " y_centers,\n", |
211 | | - " udmask.mask,\n", |
212 | | - " cmap=\"Reds_r\",\n", |
213 | | - " edgecolors=\"orange\",\n", |
214 | | - ")\n", |
215 | | - "ax1.scatter(\n", |
216 | | - " x_centers,\n", |
217 | | - " y_centers,\n", |
218 | | - " s=100,\n", |
219 | | - " c=U,\n", |
220 | | - " cmap=\"coolwarm\",\n", |
221 | | - " vmin=-0.05,\n", |
222 | | - " vmax=0.05,\n", |
223 | | - " edgecolors=\"k\",\n", |
224 | | - ")\n", |
225 | | - "ax1.quiver(\n", |
226 | | - " x_centers,\n", |
227 | | - " y_centers,\n", |
228 | | - " U,\n", |
229 | | - " V,\n", |
230 | | - " angles=\"xy\",\n", |
231 | | - " scale_units=\"xy\",\n", |
232 | | - " scale=8,\n", |
233 | | - " color=\"w\",\n", |
234 | | - " edgecolor=\"k\",\n", |
235 | | - ")\n", |
236 | | - "\n", |
237 | | - "ax2.set_xlabel(\"Longitude\")\n", |
238 | | - "ax2.set_ylabel(\"Latitude\")\n", |
239 | | - "ax2.set_title(\n", |
240 | | - " \"B) Node at center cell\\nNot according to Parcels interpolation\",\n", |
241 | | - " fontsize=14,\n", |
242 | | - " fontweight=\"bold\",\n", |
243 | | - ")\n", |
244 | | - "ax2.pcolormesh(\n", |
245 | | - " x_outcorners,\n", |
246 | | - " y_outcorners,\n", |
247 | | - " umask.mask,\n", |
248 | | - " cmap=\"Reds_r\",\n", |
249 | | - " edgecolors=\"k\",\n", |
250 | | - " linewidth=1,\n", |
251 | | - ")\n", |
252 | | - "ax2.scatter(\n", |
253 | | - " x_centers,\n", |
254 | | - " y_centers,\n", |
255 | | - " s=100,\n", |
256 | | - " c=U,\n", |
257 | | - " cmap=\"coolwarm\",\n", |
258 | | - " vmin=-0.05,\n", |
259 | | - " vmax=0.05,\n", |
260 | | - " edgecolors=\"k\",\n", |
261 | | - ")\n", |
262 | | - "ax2.quiver(\n", |
263 | | - " x_centers,\n", |
264 | | - " y_centers,\n", |
265 | | - " U,\n", |
266 | | - " V,\n", |
267 | | - " angles=\"xy\",\n", |
268 | | - " scale_units=\"xy\",\n", |
269 | | - " scale=8,\n", |
270 | | - " color=\"w\",\n", |
271 | | - " edgecolor=\"k\",\n", |
272 | | - ")\n", |
273 | | - "\n", |
274 | | - "ax3.set_xlabel(\"Longitude\")\n", |
275 | | - "ax3.set_ylabel(\"Latitude\")\n", |
276 | | - "ax3.set_title(\n", |
277 | | - " \"C) Different boundaries\\nParcels interpolates according to orange grid\",\n", |
278 | | - " fontsize=14,\n", |
279 | | - " fontweight=\"bold\",\n", |
280 | | - ")\n", |
281 | | - "ax3.pcolormesh(\n", |
282 | | - " x_centers,\n", |
283 | | - " y_centers,\n", |
284 | | - " udmask.mask,\n", |
285 | | - " cmap=\"Reds_r\",\n", |
286 | | - " edgecolors=\"orange\",\n", |
287 | | - ")\n", |
288 | | - "ax3.pcolormesh(\n", |
289 | | - " x_outcorners,\n", |
290 | | - " y_outcorners,\n", |
291 | | - " cells,\n", |
292 | | - " cmap=my_cmap,\n", |
293 | | - " edgecolors=\"black\",\n", |
294 | | - ")\n", |
295 | | - "ax3.scatter(\n", |
296 | | - " x_centers,\n", |
297 | | - " y_centers,\n", |
298 | | - " s=100,\n", |
299 | | - " c=U,\n", |
300 | | - " cmap=\"coolwarm\",\n", |
301 | | - " vmin=-0.05,\n", |
302 | | - " vmax=0.05,\n", |
303 | | - " edgecolors=\"k\",\n", |
304 | | - ")\n", |
305 | | - "ax3.quiver(\n", |
306 | | - " x_centers,\n", |
307 | | - " y_centers,\n", |
308 | | - " U,\n", |
309 | | - " V,\n", |
310 | | - " angles=\"xy\",\n", |
311 | | - " scale_units=\"xy\",\n", |
312 | | - " scale=8,\n", |
313 | | - " color=\"w\",\n", |
314 | | - " edgecolor=\"k\",\n", |
315 | | - ")\n", |
316 | | - "plt.show()" |
317 | | - ] |
318 | | - }, |
319 | 190 | { |
320 | 191 | "cell_type": "code", |
321 | 192 | "execution_count": null, |
|
0 commit comments