add saving

Author: keflavich

tutorials/position-velocity-diagrams/PVDiagramPlotting.ipynb

@@ -2,9 +2,10 @@
  "cells": [
   {
    "cell_type": "markdown",
+   "id": "ec28e7e2",
    "metadata": {},
    "source": [
-    "# Extracting and Plotting Position-Velocity Diagrams\n",
+    "fig.savefig(\"saved_pvdiagram.png\", bbox_inches='tight')fig.savefig(\"saved_pvdiagram.png\", bbox_inches='tight')fig.savefig(\"saved_pvdiagram.png\", bbox_inches='tight')fig.savefig(\"saved_pvdiagram.png\", bbox_inches='tight')fig.savefig(\"saved_pvdiagram.png\", bbox_inches='tight')fig.savefig(\"saved_pvdiagram.png\", bbox_inches='tight')fig.savefig(\"saved_pvdiagram.png\", bbox_inches='tight')fig.savefig(\"saved_pvdiagram.png\", bbox_inches='tight')# Extracting and Plotting Position-Velocity Diagrams\n",
     "\n",
     "## Authors\n",
     "Adam Ginsburg, Eric Koch\n",
@@ -25,6 +26,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "5b7f1268",
    "metadata": {},
    "source": [
     "### Header material\n",
@@ -35,6 +37,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "9f9d06e1",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -54,6 +57,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "d44ca9f9",
    "metadata": {},
    "source": [
     "Retrieve and open a cube from astropy-data:"
@@ -62,6 +66,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "f51a1376",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -71,6 +76,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "b205d98a",
    "metadata": {},
    "source": [
     "We show a single channel from the cube to visualize the data spatially.\n",
@@ -80,6 +86,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "7b2db3eb",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -88,6 +95,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "9e946c8c",
    "metadata": {},
    "source": [
     "# PV Extraction from Pixel Coordinates\n",
@@ -100,6 +108,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "7cc19bc3",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -108,6 +117,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "5e83296d",
    "metadata": {},
    "source": [
     "Then we can overplot it on our figure, now with WCS shown.  The plotting uses [WCSAxes](https://docs.astropy.org/en/stable/visualization/wcsaxes/index.html)"
@@ -116,6 +126,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "1b5e2815",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -126,13 +137,15 @@
   },
   {
    "cell_type": "markdown",
+   "id": "b0d00e33",
    "metadata": {},
    "source": [
     "`spacing` gives the separation between these points in pixels; we finely sampled by picking one-pixel spacing."
    ]
   },
   {
    "cell_type": "markdown",
+   "id": "cea5c40f",
    "metadata": {},
    "source": [
     "We can then extract the pv diagram, specifying the same spacing."
@@ -141,6 +154,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "99297938",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -150,6 +164,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "65b62bb8",
    "metadata": {},
    "source": [
     "and plot it.  `pvdiagram` is a `PrimaryHDU` object, so we need to grab the data separately from the header and convert the header to a WCS object:"
@@ -158,6 +173,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "9c602ebf",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -172,6 +188,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "e4179ebc",
    "metadata": {},
    "source": [
     "Changing units to the more commonly used km/s and more readable arcminutes can be done with [wcsaxes tools](https://docs.astropy.org/en/stable/visualization/wcsaxes/controlling_axes.html):"
@@ -180,6 +197,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "9bebb46f",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -201,6 +219,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "ec773f53",
    "metadata": {},
    "source": [
     "We can put all this together:"
@@ -209,6 +228,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "ea8c2fd2",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -221,6 +241,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "4a3ba59a",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -245,13 +266,15 @@
   },
   {
    "cell_type": "markdown",
+   "id": "f04373b1",
    "metadata": {},
    "source": [
     "# PV Extraction from Sky Coordinates"
    ]
   },
   {
    "cell_type": "markdown",
+   "id": "f7643042",
    "metadata": {},
    "source": [
     "We can also make paths using celestial coordinates by passing coordinates defined in an`~astropy.coordinates.SkyCoord` object to `~pvextractor.Path`."
@@ -260,6 +283,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "4f0f1d40",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -269,6 +293,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "b3299ae2",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -277,6 +302,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "27c2224d",
    "metadata": {},
    "source": [
     "We can plot again; the coordinates will be automatically determined"
@@ -285,6 +311,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "2d3f42c3",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -296,6 +323,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "65c46371",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -306,6 +334,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "cbfa2a87",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -330,6 +359,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "2bca4a13",
    "metadata": {},
    "source": [
     "We can also change the aspect ratio of the PV diagram:"
@@ -338,6 +368,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "fcf34393",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -363,13 +394,15 @@
   },
   {
    "cell_type": "markdown",
+   "id": "e8deae20",
    "metadata": {},
    "source": [
     "## PV Extraction with Spatial Averaging"
    ]
   },
   {
    "cell_type": "markdown",
+   "id": "0824ae9d",
    "metadata": {},
    "source": [
     "`~pvextractor.Path` allows you to specify a `width` to average over, which specifies a spatial range around the path to average over:"
@@ -378,6 +411,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "91f5420f",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -387,6 +421,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "bb057443",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -395,6 +430,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "1cf3ccc5",
    "metadata": {},
    "source": [
     "We can plot this path as a set of patches to show where we averaged.  The default spacing is 1 pixel,so we plot 1-pixel chunks."
@@ -403,6 +439,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "9a866a1f",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -434,6 +471,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "d70082ef",
    "metadata": {},
    "source": [
     "We can also have more widely spaced chunks.\n",
@@ -444,6 +482,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "98faa709",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -475,6 +514,7 @@
   },
   {
    "cell_type": "markdown",
+   "id": "c8203c9d",
    "metadata": {},
    "source": [
     "## Saving\n",
@@ -485,11 +525,31 @@
   {
    "cell_type": "code",
    "execution_count": null,
+   "id": "d59d2ff3",
    "metadata": {},
    "outputs": [],
    "source": [
     "pvdiagram.writeto(\"saved_pvdiagram.fits\", overwrite=True)"
    ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "560c2f93-ee95-4d60-825a-83ec9f3acba7",
+   "metadata": {},
+   "source": [
+    "We can also save the figure as a png or pdf:"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "74de9f99-e2fd-411f-9249-a39fa0b35908",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "fig.savefig(\"saved_pvdiagram.png\", bbox_inches='tight')\n",
+    "fig.savefig(\"saved_pvdiagram.pdf\", bbox_inches='tight')"
+   ]
   }
  ],
  "metadata": {