-
Notifications
You must be signed in to change notification settings - Fork 11
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge branch 'main' into maint/readme-python-example-test
- Loading branch information
Showing
17 changed files
with
527 additions
and
206 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -7,3 +7,4 @@ chapters: | |
| - file: background | ||
| - file: design | ||
| - file: creating_the_graph | ||
| - file: lat_lons | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,180 @@ | ||
| { | ||
| "cells": [ | ||
| { | ||
| "cell_type": "markdown", | ||
| "id": "65f79439-0b75-483f-858b-932422f7599d", | ||
| "metadata": {}, | ||
| "source": [ | ||
| "# Working with lat-lon coordinates\n", | ||
| "\n", | ||
| "In the previous sections we have considered grid point positions `coords` given as Cartesian coordinates. However, it is common that we have coordinates given as latitudes and longitudes. This notebook describes how we can constuct graphs directly using lat-lon coordinates. This is achieved by specifying the Coordinate Reference System (CRS) of `coords` and the CRS that the graph construction should be carried out in. `coords` will then be projected to this new CRS before any calculations are carried out." | ||
| ] | ||
| }, | ||
| { | ||
| "cell_type": "markdown", | ||
| "id": "8cbcd3bd-a80c-41e4-a54b-bcfb5dfbb75c", | ||
| "metadata": {}, | ||
| "source": [ | ||
| "## A motivating example" | ||
| ] | ||
| }, | ||
| { | ||
| "cell_type": "markdown", | ||
| "id": "4c710a27-7e03-4f4f-a4dd-5d1c53a1e4eb", | ||
| "metadata": {}, | ||
| "source": [ | ||
| "Let's start by defining some example lat-lons to use in our example. When using lat-lons the first column of `coords` should contain longitudes and the second column latitudes.\n", | ||
| "\n", | ||
| "In the example below we create lat-lons laid out around the geographic North Pole. These example points are equidistantly spaced, but this does not have to be the case." | ||
| ] | ||
| }, | ||
| { | ||
| "cell_type": "code", | ||
| "execution_count": null, | ||
| "id": "9d577c2f-6dca-4b1b-8bdc-1359e6573cda", | ||
| "metadata": {}, | ||
| "outputs": [], | ||
| "source": [ | ||
| "import numpy as np\n", | ||
| "import matplotlib.pyplot as plt\n", | ||
| "import cartopy.crs as ccrs\n", | ||
| "\n", | ||
| "longitudes = np.linspace(-180, 180, 40)\n", | ||
| "latitudes = np.linspace(65, 85, 5) # Very close to north pole\n", | ||
| "\n", | ||
| "meshgridded_lat_lons = np.meshgrid(longitudes, latitudes)\n", | ||
| "coords = np.stack([mg_coord.flatten() for mg_coord in meshgridded_lat_lons], axis=1)\n", | ||
| "\n", | ||
| "fig, ax = plt.subplots(figsize=(15, 9), subplot_kw={\"projection\": ccrs.PlateCarree()})\n", | ||
| "ax.scatter(coords[:, 0], coords[:, 1], marker=\".\")\n", | ||
| "ax.coastlines()\n", | ||
| "ax.set_extent((-180, 180, -90, 90))" | ||
| ] | ||
| }, | ||
| { | ||
| "cell_type": "markdown", | ||
| "id": "4d4f5f35-01bf-4eeb-be52-d7deabbf2039", | ||
| "metadata": {}, | ||
| "source": [ | ||
| "We first consider what happens if we directly feed these lat-lons as `coords`, treating them as if they were Cartesian coordinates. In this notebook we will only create flat \"Keisler-like\" graphs, but everything works analogously for the other graph types." | ||
| ] | ||
| }, | ||
| { | ||
| "cell_type": "code", | ||
| "execution_count": null, | ||
| "id": "8e04abfd-f3e2-4f77-908c-6e2374431e9e", | ||
| "metadata": {}, | ||
| "outputs": [], | ||
| "source": [ | ||
| "import weather_model_graphs as wmg\n", | ||
| "\n", | ||
| "graph = wmg.create.archetype.create_keisler_graph(coords, mesh_node_distance=10)\n", | ||
| "fig, ax = plt.subplots(figsize=(15, 9), subplot_kw={\"projection\": ccrs.PlateCarree()})\n", | ||
| "wmg.visualise.nx_draw_with_pos_and_attr(\n", | ||
| " graph, ax=ax, node_size=30, edge_color_attr=\"component\", node_color_attr=\"type\"\n", | ||
| ")\n", | ||
| "ax.coastlines()\n", | ||
| "ax.set_extent((-180, 180, -90, 90))" | ||
| ] | ||
| }, | ||
| { | ||
| "cell_type": "markdown", | ||
| "id": "65b39a86-d64f-46a8-a732-0e608a3fb07f", | ||
| "metadata": {}, | ||
| "source": [ | ||
| "This creates a useable mesh graph, but we can note a few problems with it:\n", | ||
| "\n", | ||
| "* There are no connections between nodes around longitude -180/180, i.e. the periodicity of longitude is not considered.\n", | ||
| "* All nodes at the top of the plot, close to the pole, are actually very close spatially. Yet there are no connections between them.\n", | ||
| "\n", | ||
| "These are issues both in the connection between the grid nodes and the mesh, and in the connections between mesh nodes. This points to the fact that we should probably use a different projection when building our graph. " | ||
| ] | ||
| }, | ||
| { | ||
| "cell_type": "markdown", | ||
| "id": "103a3784-6ad4-4188-ba02-189f9cf71b2f", | ||
| "metadata": {}, | ||
| "source": [ | ||
| "## Constructing a graph within a projection\n", | ||
| "For our example above, let's instead try to construct the graph based on first projecting our lat-lon coordinates to another CRS with 2-dimensional cartesian coordinates. This can be done by giving the `coords_crs` and `graph_crs` arguments to the graph creation functions. Theses arguments should both be instances of `pyproj.crs.CRS` ([pyproj docs.](https://pyproj4.github.io/pyproj/stable/api/crs/crs.html#pyproj.crs.CRS)). Nicely, they can be `cartopy.crs.CRS`, which provides easy ways to specify such CRSs. For more advanced use cases a `pyproj.crs.CRS` can be specified directly. See [the cartopy documentation](https://scitools.org.uk/cartopy/docs/latest/reference/projections.html) for a list of readily available CRSs to use for projecting the coordinates. \n", | ||
| "\n", | ||
| "We will here try the same thing as above, but using a Azimuthal equidistant projection centered at the pole. The CRS of our lat-lon coordinates will be `cartopy.crs.PlateCarree` and we want to project this to `cartopy.crs.AzimuthalEquidistant`:" | ||
| ] | ||
| }, | ||
| { | ||
| "cell_type": "code", | ||
| "execution_count": null, | ||
| "id": "fffd518b-f323-4e24-8544-60edeaaa8bb2", | ||
| "metadata": {}, | ||
| "outputs": [], | ||
| "source": [ | ||
| "# Define our projection\n", | ||
| "coords_crs = ccrs.PlateCarree()\n", | ||
| "graph_crs = ccrs.AzimuthalEquidistant(central_latitude=90)\n", | ||
| "\n", | ||
| "fig, ax = plt.subplots(figsize=(15, 9), subplot_kw={\"projection\": graph_crs})\n", | ||
| "ax.scatter(coords[:, 0], coords[:, 1], marker=\".\", transform=ccrs.PlateCarree())\n", | ||
| "_ = ax.coastlines()" | ||
| ] | ||
| }, | ||
| { | ||
| "cell_type": "markdown", | ||
| "id": "afa6b933-121b-41c4-a341-9e548b92ad87", | ||
| "metadata": {}, | ||
| "source": [ | ||
| "Note that distances within projections tend to have very large magnitudes, so the distance between mesh nodes should be specified accordingly." | ||
| ] | ||
| }, | ||
| { | ||
| "cell_type": "code", | ||
| "execution_count": null, | ||
| "id": "1ddd7ec3-0d66-4feb-a0d3-293f3194dddd", | ||
| "metadata": {}, | ||
| "outputs": [], | ||
| "source": [ | ||
| "mesh_distance = (\n", | ||
| " 10**6\n", | ||
| ") # Large euclidean distance in projection coordinates between mesh nodes\n", | ||
| "graph = wmg.create.archetype.create_keisler_graph(\n", | ||
| " coords, mesh_node_distance=mesh_distance, coords_crs=coords_crs, graph_crs=graph_crs\n", | ||
| ") # Note that we here specify the projection argument\n", | ||
| "fig, ax = plt.subplots(figsize=(15, 9), subplot_kw={\"projection\": graph_crs})\n", | ||
| "wmg.visualise.nx_draw_with_pos_and_attr(\n", | ||
| " graph, ax=ax, node_size=30, edge_color_attr=\"component\", node_color_attr=\"type\"\n", | ||
| ")\n", | ||
| "_ = ax.coastlines()" | ||
| ] | ||
| }, | ||
| { | ||
| "cell_type": "markdown", | ||
| "id": "f5acf83a-df33-4925-bb32-c106e27e51a4", | ||
| "metadata": {}, | ||
| "source": [ | ||
| "Now this looks like a more reasonable graph layout, that better respects the spatial relations between the grid points. There are still things that could be tweaked further (e.g. the large number of grid nodes connected to the center mesh node), but this ends our example of defining graphs using lat-lon coordinates.\n", | ||
| "\n", | ||
| "It can be noted that this projection between different CRSs provides more general functionality than just handling lat-lon coordinates. It is entirely possible to transform from any `coords_crs` to any `graph_crs` using these arguments." | ||
| ] | ||
| } | ||
| ], | ||
| "metadata": { | ||
| "kernelspec": { | ||
| "display_name": "Python 3 (ipykernel)", | ||
| "language": "python", | ||
| "name": "python3" | ||
| }, | ||
| "language_info": { | ||
| "codemirror_mode": { | ||
| "name": "ipython", | ||
| "version": 3 | ||
| }, | ||
| "file_extension": ".py", | ||
| "mimetype": "text/x-python", | ||
| "name": "python", | ||
| "nbconvert_exporter": "python", | ||
| "pygments_lexer": "ipython3", | ||
| "version": "3.10.15" | ||
| } | ||
| }, | ||
| "nbformat": 4, | ||
| "nbformat_minor": 5 | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.