data_exploration/figure/Supplemental/verticalMovement.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 35,
   "id": "488e9afe",
   "metadata": {},
   "outputs": [],
   "source": [
    "#dataframes\n",
    "import pandas as pd\n",
    "import h5py\n",
    "\n",
    "#speedup\n",
    "import multiprocessing as mp\n",
    "\n",
    "#suppress warnings\n",
    "import warnings\n",
    "warnings.simplefilter(action='ignore', category=FutureWarning)\n",
    "pd.TimeSeries = pd.Series \n",
    "\n",
    "#math\n",
    "import numpy as np\n",
    "from scipy.spatial.distance import cdist\n",
    "from scipy import signal\n",
    "import math\n",
    "import copy\n",
    "\n",
    "\n",
    "#plots\n",
    "import pylab as plt\n",
    "import matplotlib.pyplot as plt\n",
    "from matplotlib.ticker import MaxNLocator\n",
    "import matplotlib\n",
    "from figurefirst import FigureLayout,mpl_functions\n",
    "import figurefirst\n",
    "import matplotlib.image as mpimg\n",
    "from mpl_toolkits.axes_grid1 import make_axes_locatable # for colorbar\n",
    "import seaborn as sns\n",
    "sns.set()\n",
    "sns.set_style(\"whitegrid\")\n",
    "import figurefirst as fifi\n",
    "from IPython.display import display,SVG\n",
    "\n",
    "#misc\n",
    "import cairosvg\n",
    "import itertools\n",
    "import time\n",
    "np.set_printoptions(suppress=True)\n",
    "\n",
    "FS=7"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 36,
   "id": "2b33a22b",
   "metadata": {},
   "outputs": [],
   "source": [
    "dir='~/DataAnalysis/data/DesertWind/'\n",
    "df1 = pd.read_hdf(dir+\"20201020_172919_windy_b_data.hdf\")\n",
    "# df2 = pd.read_hdf(\"../../../../Figure/methodfigure2/windgpsdata.hdf\") #Forest\n",
    "dir2 = '~/DataAnalysis/data/ForestWindData/'\n",
    "df2 = pd.read_hdf(dir2 + \"7.7.21_maskedgpsdata.hdf\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 37,
   "id": "eb9a1da2",
   "metadata": {},
   "outputs": [],
   "source": [
    "dt=df1.time_epoch.iloc[1]-df1.time_epoch.iloc[0]\n",
    "arr_dat = [df1,df2]\n",
    "for i in range(len(arr_dat)):\n",
    "    dataframe = arr_dat[i]\n",
    "    time = []\n",
    "    time.append(0)\n",
    "    for j in range(1,len(arr_dat[i])):\n",
    "        time.append(time[j-1]+dt)\n",
    "\n",
    "    arr_dat[i]['time'] = time     "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 51,
   "id": "5cebf256",
   "metadata": {
    "scrolled": false
   },
   "outputs": [
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAAASQAAADTCAYAAAAoGW75AAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjUuMSwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy/YYfK9AAAACXBIWXMAAAsTAAALEwEAmpwYAAAUcklEQVR4nO3dXUhU6R8H8K8zO1qYsWuojdXSFqsNvbEZ/6AySs20ZnI3LMOKSlKi2KAg8qJ8SYkmKCipi+2ifSsqLzIysYguzOidwMzKMsvKUUuRwF6cjs//IhyyyebscTrnUb8fECZ5lO8veb6cc5xzDBBCCBARScBkdAAiom4sJCKSBguJiKTBQiIiabCQiEgaLCQikobPQnI6nYiLi0N0dDRqa2u/uEZRFOTn5yMhIQHz589HcXGx34MS0cDns5Di4+Nx9OhRjBo1qtc1Z86cQUNDA86fP48TJ06gqKgIz58/92tQIhr4fBbS9OnTYbVav7qmrKwMS5cuhclkQmhoKBISElBeXu63kEQ0OHznj2/icrkQGRnp+bfVakVTU5Pqr+/q6kJHRwcsFgsCAgL8EYmIdCaEgNvtRnBwMEwmbZen/VJIfdXR0dHr9Ski6l+ioqIQEhKi6Wv9UkhWqxWNjY2YMmUKAO8jJl8sFguAj4MEBgb6I5IhqqurMWnSJKNjaMb8xuvPM3R2dqK2ttazn7XwSyElJSWhuLgYiYmJaG9vx4ULF3D06FHVX999mhYYGIigoCB/RDIM8xurv+cH+v8Mfbns4vNEr7CwEHPmzEFTUxPWrl2LRYsWAQAyMzNx584dAEBKSgpGjx6NxMRELFu2DBs3bsSYMWM0hyKiwcnnEdL27duxfft2r88fPnzY89psNiM/P9+/yYho0OE7tYlIGiwkIpIGC4mIpMFCIiJpsJCISBosJCKSBguJiKTBQiIiabCQiEgaLCQikgYLiYikwUIiImmwkIhIGiwkIpIGC4mIpMFCIiJpsJCISBosJCKSBguJiKTBQiIiabCQiEgaLCQikgYLiYikwUIiImmwkIhIGiwkIpIGC4mIpMFCIiJpsJCISBrfqVlUX1+P7OxstLe34/vvv4fT6cTYsWN7rCkqKsKxY8cQHh4OAJg2bRpyc3P9HpiIBi5VhZSbm4v09HSkpKTg9OnTyMnJwd9//+217tdff8W2bdv8HpKIBgefp2ytra2oqamB3W4HANjtdtTU1KCtre2bhyOiwcVnIblcLkRERMBsNgMAzGYzwsPD4XK5vNaePXsWDocDGRkZuH37tv/TEtGApuqUTY3ly5dj/fr1sFgsuHz5MjZs2ICysjL88MMPqr9HdXW1v+IY5tatW0ZH6BPmN95AmEErn4VktVrR3NwMRVFgNpuhKApaWlpgtVp7rAsLC/O8njVrFqxWKx4+fIj//e9/qsNMmjQJQUFB/yG+XG7duoWYmBijY2jG/MbrzzO8f/++zwcVPk/ZRowYAZvNhtLSUgBAaWkpbDYbQkNDe6xrbm72vL537x5evHiBn376qU/hiGhwUXXKlpeXh+zsbBw6dAjDhw+H0+kEAGRmZmLTpk2YPHky9u3bh7t378JkMsFisWDPnj09jpqIiHxRVUjjx49HcXGx1+cPHz7sed1dUkREWvGd2kQkDRYSEUmDhURE0mAhEZE0WEhEJA0WEhFJg4VERNJgIRGRNFhIRCQNFhIRSYOFRETSYCERkTRYSEQkDRYSEUmDhURE0mAhEZE0WEhEJA0WEhFJg4VERNJgIRGRNFhIRCQNFhIRSYOFRETSYCERkTRYSEQkDRYSEUmDhURE0mAhEZE0WEhEJA1VhVRfX4+0tDQsWLAAaWlpePLkidcaRVGQn5+PhIQEzJ8/H8XFxf7OSkQDnKpCys3NRXp6Os6dO4f09HTk5OR4rTlz5gwaGhpw/vx5nDhxAkVFRXj+/LnfAxPRwOWzkFpbW1FTUwO73Q4AsNvtqKmpQVtbW491ZWVlWLp0KUwmE0JDQ5GQkIDy8vJvk5qIBqTvfC1wuVyIiIiA2WwGAJjNZoSHh8PlciE0NLTHusjISM+/rVYrmpqaVIUQQgAAOjs7/1N4Gb1//97oCH3C/MbrrzN079/u/ayFz0LSg9vtBgDU1tYanKTvqqurjY7QJ8xvvP4+g9vtxpAhQzR9rc9CslqtaG5uhqIoMJvNUBQFLS0tsFqtXusaGxsxZcoUAN5HTF8THByMqKgoWCwWBAQEaBiDiIwmhIDb7UZwcLDm7+GzkEaMGAGbzYbS0lKkpKSgtLQUNputx+kaACQlJaG4uBiJiYlob2/HhQsXcPToUVUhTCYTQkJCtE1ARNLQemTULUCoOOGrq6tDdnY2Xr9+jeHDh8PpdGLcuHHIzMzEpk2bMHnyZCiKgp07d+Ly5csAgMzMTKSlpfUpHBENLqoKiYhID3ynNhFJg4VERNJgIRGRNFhIRCQNFhIRSUPXQhoITw1QM8PBgwexaNEiOBwOLFmyBJcuXdI/aC/U5O/2+PFjTJ06FU6nU7+AKqidoaysDA6HA3a7HQ6HA69evdI3aC/U5G9tbUVWVhYcDgeSk5ORl5eHDx8+6B/2C5xOJ+Li4hAdHd3r3RWa97HQ0apVq0RJSYkQQoiSkhKxatUqrzWnTp0SGRkZQlEU0draKmJjY8WzZ8/0jPlVamaoqKgQb968EUIIce/ePRETEyPevn2ra87eqMkvhBAfPnwQK1euFFu2bBG7d+/WM6JPamaoqqoSycnJoqWlRQghxOvXr8W7d+90zdkbNfkLCws9/++dnZ0iNTVVnD17Vtecvblx44ZobGwU8+bNEw8ePPjiGq37WLcjpIHw1AC1M8TGxmLo0KEAgOjoaAgh0N7erndcL2rzA8Aff/yBuXPnYuzYsTqn/Dq1M/z555/IyMhAWFgYACAkJARBQUG65/2c2vwBAQHo6OhAV1cXOjs74Xa7ERERYURkL9OnT/e6dexzWvexboX0tacGfL5O61MDvjW1M3yqpKQEP/74I0aOHKlXzF6pzX///n1UVlZizZo1BqT8OrUz1NXV4dmzZ1ixYgV+++03HDp0qE93ofuL2vwbNmxAfX09Zs+e7fmIiYkxIrImWvcxL2p/Q9evX8f+/fuxd+9eo6Oo5na7sWPHDuTn53s2TX+kKAoePHiAI0eO4J9//kFFRQVOnz5tdCzVysvLER0djcrKSlRUVODmzZvSnCl8S7oV0qdPDQDg86kB3VwulxRHF4D6GQDg9u3b2Lp1Kw4ePIhx48bpHfWL1OR/+fIlGhoakJWVhbi4OPz11184efIkduzYYVTsHtT+DCIjI5GUlITAwEAMGzYM8fHxqKqqMiJyD2rz//vvv1i8eLHnxvO4uDhcu3bNiMiaaN3HuhXSp08NAODzqQFdXV1oa2vDhQsXsGDBAr1ifpXaGaqqqrB582YcOHAAEydONCLqF6nJHxkZiWvXruHixYu4ePEiVq9ejWXLlqGgoMCo2D2o/RnY7XZUVlZ6Holx9epVTJgwwYjIPajNP3r0aFRUVAD4+OCzK1eu4Oeff9Y9r1aa97FfL7/78OjRI5GamioSExNFamqqqKurE0IIsW7dOlFVVSWE+PjbnZycHBEfHy/i4+PF8ePH9Yzok5oZlixZImbMmCEWL17s+bh//76RsT3U5P/UgQMHpPstm5oZFEURu3btEklJSWLhwoVi165dQlEUI2N7qMn/9OlTsWbNGmG320VycrLIy8sTbrfbyNgeBQUFIjY2VthsNjFz5kyxcOFCIYR/9rHPu/2dTifOnTuHFy9e4MyZM4iKivJaoygKCgsLcenSJQQEBCArKwtLly7V3K5ENDj5PGWLj4/H0aNHMWrUqF7X8C+OEJE/+Hxi5PTp031+k97ec7Bu3TpVIbq6utDR0cFH2BL1Y+KTR9iaTNouT/vlIf99fe9QR0fHgHjAPxEBUVFRmh9JLcVfHbFYLAA+DhIYGGhwGu2qq6sxadIko2NoxvzG688zdHZ2ora21rOftfBLIfXlL44A8JymBQYGSvH2/r5gfmP19/xA/5+hL5dd/PI+JJnfO0RE/YfPQiosLMScOXPQ1NSEtWvXYtGiRQA+/lWRO3fuAABSUlIwevRoJCYmYtmyZdi4cSPGjBnzbZMT0YDj85Rt+/bt2L59u9fnDx8+7HltNpuRn5/v32RENOjw5loikgYLiYikwUIiImmwkIhIGiwkIpIGC4mIpMFCIiJpsJCISBosJCKSBguJiKTBQiIiabCQiEgaLCQikgYLiYikwUIiImmwkIhIGiwkIpIGC4mIpMFCIiJpsJCISBosJCKSBguJiKTBQiIiabCQiEgaLCQikgYLiYikwUIiImmwkIhIGiwkIpLGd2oW1dfXIzs7G+3t7fj+++/hdDoxduzYHmuKiopw7NgxhIeHAwCmTZuG3NxcvwcmooFLVSHl5uYiPT0dKSkpOH36NHJycvD33397rfv111+xbds2v4ckosHB5ylba2srampqYLfbAQB2ux01NTVoa2v75uGIaHDxeYTkcrkQEREBs9kMADCbzQgPD4fL5UJoaGiPtWfPnkVlZSXCwsLw+++/45dffvlPYaqrq//TehndunXL6Ah9wvzGGwgzaKXqlE2N5cuXY/369bBYLLh8+TI2bNiAsrIy/PDDD6q/x6RJkxAUFOSvSLq7desWYmJijI6hGfMbrz/P8P79+z4fVPg8ZbNarWhuboaiKAAARVHQ0tICq9XaY11YWBgsFgsAYNasWbBarXj48GGfwhHR4OKzkEaMGAGbzYbS0lIAQGlpKWw2m9fpWnNzs+f1vXv38OLFC/z0009+jktEA5mqU7a8vDxkZ2fj0KFDGD58OJxOJwAgMzMTmzZtwuTJk7Fv3z7cvXsXJpMJFosFe/bsQVhY2DcNT0QDi6pCGj9+PIqLi70+f/jwYc/r7pIiItKK79QmImmwkIhIGiwkIpIGC4mIpMFCIiJpsJCISBosJCKSBguJiKTBQiIiabCQiEgaLCQikgYLiYikwUIiImmwkIhIGiwkIpIGC4mIpMFCIiJpsJCISBosJCKSBguJiKTBQiIiabCQiEgaLCQikgYLiYikwUIiImmwkIhIGiwkIpIGC4mIpMFCIiJpqCqk+vp6pKWlYcGCBUhLS8OTJ0+81iiKgvz8fCQkJGD+/PkoLi72d1YiGuBUFVJubi7S09Nx7tw5pKenIycnx2vNmTNn0NDQgPPnz+PEiRMoKirC8+fP/R6YiAau73wtaG1tRU1NDY4cOQIAsNvtKCgoQFtbG0JDQz3rysrKsHTpUphMJoSGhiIhIQHl5eVYt26dzxBCCABAZ2en1jmk8f79e6Mj9AnzG6+/ztC9f7v3sxY+C8nlciEiIgJmsxkAYDabER4eDpfL1aOQXC4XIiMjPf+2Wq1oampSFcLtdgMAamtr/1N4GVVXVxsdoU+Y33j9fQa3240hQ4Zo+lqfhaSH4OBgREVFwWKxICAgwOg4RKSBEAJutxvBwcGav4fPQrJarWhuboaiKDCbzVAUBS0tLbBarV7rGhsbMWXKFADeR0xfYzKZEBISoiE+EclE65FRN58XtUeMGAGbzYbS0lIAQGlpKWw2W4/TNQBISkpCcXExurq60NbWhgsXLmDBggV9CkdEg0uAUHEFqq6uDtnZ2Xj9+jWGDx8Op9OJcePGITMzE5s2bcLkyZOhKAp27tyJy5cvAwAyMzORlpb2zQcgooFDVSEREemB79QmImmwkIhIGiwkIpIGC4mIpKFrIQ2Em3TVzHDw4EEsWrQIDocDS5YswaVLl/QP2gs1+bs9fvwYU6dOhdPp1C+gCmpnKCsrg8PhgN1uh8PhwKtXr/QN2gs1+VtbW5GVlQWHw4Hk5GTk5eXhw4cP+of9AqfTibi4OERHR/d6d4XmfSx0tGrVKlFSUiKEEKKkpESsWrXKa82pU6dERkaGUBRFtLa2itjYWPHs2TM9Y36VmhkqKirEmzdvhBBC3Lt3T8TExIi3b9/qmrM3avILIcSHDx/EypUrxZYtW8Tu3bv1jOiTmhmqqqpEcnKyaGlpEUII8fr1a/Hu3Ttdc/ZGTf7CwkLP/3tnZ6dITU0VZ8+e1TVnb27cuCEaGxvFvHnzxIMHD764Rus+1u0IqfsmXbvdDuDjTbo1NTVoa2vrsa63m3RloHaG2NhYDB06FAAQHR0NIQTa29v1jutFbX4A+OOPPzB37lyMHTtW55Rfp3aGP//8ExkZGQgLCwMAhISEICgoSPe8n1ObPyAgAB0dHejq6kJnZyfcbjciIiKMiOxl+vTpXndqfE7rPtatkL52k+7n67TepPutqZ3hUyUlJfjxxx8xcuRIvWL2Sm3++/fvo7KyEmvWrDEg5depnaGurg7Pnj3DihUr8Ntvv+HQoUN9ugvdX9Tm37BhA+rr6zF79mzPR0xMjBGRNdG6j3lR+xu6fv069u/fj7179xodRTW3240dO3YgPz/fs2n6I0VR8ODBAxw5cgT//PMPKioqcPr0aaNjqVZeXo7o6GhUVlaioqICN2/elOZM4VvSrZA+vUkXgM+bdLu5XC4pji4A9TMAwO3bt7F161YcPHgQ48aN0zvqF6nJ//LlSzQ0NCArKwtxcXH466+/cPLkSezYscOo2D2o/RlERkYiKSkJgYGBGDZsGOLj41FVVWVE5B7U5v/333+xePFiz43ncXFxuHbtmhGRNdG6j3UrpIFwk67aGaqqqrB582YcOHAAEydONCLqF6nJHxkZiWvXruHixYu4ePEiVq9ejWXLlqGgoMCo2D2o/RnY7XZUVlZ6Holx9epVTJgwwYjIPajNP3r0aFRUVAD4+OCzK1eu4Oeff9Y9r1aa97FfL7/78OjRI5GamioSExNFamqqqKurE0IIsW7dOlFVVSWE+PjbnZycHBEfHy/i4+PF8ePH9Yzok5oZlixZImbMmCEWL17s+bh//76RsT3U5P/UgQMHpPstm5oZFEURu3btEklJSWLhwoVi165dQlEUI2N7qMn/9OlTsWbNGmG320VycrLIy8sTbrfbyNgeBQUFIjY2VthsNjFz5kyxcOFCIYR/9jFvriUiafCiNhFJg4VERNJgIRGRNFhIRCQNFhIRSYOFRETSYCERkTRYSEQkjf8DR5NAuae25AgAAAAASUVORK5CYII=\n",
      "text/plain": [
       "<Figure size 595.276x841.89 with 2 Axes>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "layout = fifi.svg_to_axes.FigureLayout('../../../../Figure/verticalMovement.svg',\n",
    "                                           autogenlayers=True, make_mplfigures=True, hide_layers=[])"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "465bc0b6",
   "metadata": {},
   "source": [
    "## W Movement Desert Data"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 52,
   "id": "e0b0c7ea",
   "metadata": {},
   "outputs": [],
   "source": [
    "ax=layout.axes[('fig','ax1')]\n",
    "ax.grid(False)\n",
    "line_width=0.5\n",
    "marker_size=0.5\n",
    "\n",
    "ax.plot(df1.time,df1.W,linewidth=line_width,markersize=marker_size*2,rasterized= True)\n",
    "# ax.set_xlim(200,800)\n",
    "# ax.set_ylim(-1.5,1.5)\n",
    "\n",
    "ax.set_rasterization_zorder(-20)\n",
    "\n",
    "mpl_functions.adjust_spines(ax,['left','bottom'],\n",
    "                            spine_locations={'left':4,'bottom':4},\n",
    "                            smart_bounds=True,\n",
    "                            yticks=[-1.5,1.5],\n",
    "                            xticks=[0,80000],\n",
    "                            linewidth=0.5)\n",
    "\n",
    "ax.set_xticklabels(['0','6'])\n",
    "ax.set_ylabel('Ambient Wind\\n Speed, m/s')\n",
    "ax.yaxis.set_label_coords(-0.04, 0.5)\n",
    "\n",
    "figurefirst.mpl_functions.set_fontsize(ax, FS)\n",
    "layout.append_figure_to_layer(layout.figures['fig'], 'fig', cleartarget=True)\n",
    "layout.write_svg('../../../../Figure/verticalMovement.svg')"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "b95f00ee",
   "metadata": {},
   "source": [
    "## W Movement Forest Data"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 53,
   "id": "142ab7d9",
   "metadata": {},
   "outputs": [],
   "source": [
    "ax=layout.axes[('fig','ax2')]\n",
    "ax.grid(False)\n",
    "line_width=0.5\n",
    "marker_size=0.5\n",
    "\n",
    "ax.plot(df2.time,df2.W_1,linewidth=line_width,markersize=marker_size*2,rasterized= True)\n",
    "ax.set_rasterization_zorder(-20)\n",
    "\n",
    "mpl_functions.adjust_spines(ax,['left','bottom'],\n",
    "                            spine_locations={'left':4,'bottom':4},\n",
    "                            smart_bounds=True,\n",
    "                            yticks=[-1.5,1.5],\n",
    "                            xticks=[0,50000],\n",
    "                            linewidth=0.5)\n",
    "\n",
    "ax.set_xticklabels(['0','4'])\n",
    "ax.set_xlabel('Time, hr')\n",
    "ax.xaxis.set_label_coords(0.5, -0.1)\n",
    "ax.set_ylabel('Ambient Wind\\n Speed, m/s')\n",
    "ax.yaxis.set_label_coords(-0.04, 0.5)\n",
    "\n",
    "figurefirst.mpl_functions.set_fontsize(ax, FS)\n",
    "layout.append_figure_to_layer(layout.figures['fig'], 'fig', cleartarget=True)\n",
    "layout.write_svg('../../../../Figure/verticalMovement.svg')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "337bb1f6",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "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.8.10"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}