From f818e662a8fbc101d575bbd169f818a98d6fa9fd Mon Sep 17 00:00:00 2001 From: David A Minton Date: Thu, 12 Aug 2021 12:42:45 -0400 Subject: [PATCH] Updated example for Swifter/Swiftest comparison of SyMBA --- .../swiftest_vs_swifter.ipynb | 24 +- .../mars_disk/param.swifter.in | 2 +- .../mars_disk/param.swiftest.in | 2 +- .../mars_disk/swiftest_vs_swifter.ipynb | 557 ++++++++++++++++++ 4 files changed, 571 insertions(+), 14 deletions(-) create mode 100644 examples/symba_swifter_comparison/mars_disk/swiftest_vs_swifter.ipynb diff --git a/examples/symba_swifter_comparison/1pl_1pl_encounter/swiftest_vs_swifter.ipynb b/examples/symba_swifter_comparison/1pl_1pl_encounter/swiftest_vs_swifter.ipynb index dee66b3b6..ec0e145ef 100644 --- a/examples/symba_swifter_comparison/1pl_1pl_encounter/swiftest_vs_swifter.ipynb +++ b/examples/symba_swifter_comparison/1pl_1pl_encounter/swiftest_vs_swifter.ipynb @@ -81,8 +81,8 @@ { "data": { "text/plain": [ - "[,\n", - " ]" + "[,\n", + " ]" ] }, "execution_count": 6, @@ -108,7 +108,7 @@ }, { "cell_type": "code", - "execution_count": 14, + "execution_count": 7, "metadata": {}, "outputs": [ { @@ -509,7 +509,7 @@ " -3.40231310e-09])\n", "Coordinates:\n", " id float64 2.0\n", - " * time (y) (time (y)) float64 0.0 0.0006845 0.001369 ... 0.1492 0.1499 0.1506
    • id
      ()
      float64
      2.0
      array(2.)
    • time (y)
      (time (y))
      float64
      0.0 0.0006845 ... 0.1499 0.1506
      array([0.      , 0.000684, 0.001369, ..., 0.149213, 0.149897, 0.150582])
    • " ], "text/plain": [ "\n", @@ -599,7 +599,7 @@ " * time (y) (time (y)) float64 0.0 0.0006845 0.001369 ... 0.1492 0.1499 0.1506" ] }, - "execution_count": 14, + "execution_count": 7, "metadata": {}, "output_type": "execute_result" } @@ -610,7 +610,7 @@ }, { "cell_type": "code", - "execution_count": 11, + "execution_count": 8, "metadata": {}, "outputs": [ { @@ -987,7 +987,7 @@ " nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, nan])\n", "Coordinates:\n", " id float64 100.0\n", - " * time (y) (time (y)) float64 0.0 0.0006845 0.001369 ... 0.1492 0.1499 0.1506
    • id
      ()
      float64
      100.0
      array(100.)
    • time (y)
      (time (y))
      float64
      0.0 0.0006845 ... 0.1499 0.1506
      array([0.      , 0.000684, 0.001369, ..., 0.149213, 0.149897, 0.150582])
    • " ], "text/plain": [ "\n", @@ -1029,7 +1029,7 @@ " * time (y) (time (y)) float64 0.0 0.0006845 0.001369 ... 0.1492 0.1499 0.1506" ] }, - "execution_count": 11, + "execution_count": 8, "metadata": {}, "output_type": "execute_result" } @@ -1441,7 +1441,7 @@ " nan])\n", "Coordinates:\n", " id int64 100\n", - " * time (time) float64 0.0 0.0006845 0.001369 ... 0.1492 0.1499 0.1506
    • id
      ()
      int64
      100
      array(100)
    • time
      (time)
      float64
      0.0 0.0006845 ... 0.1499 0.1506
      array([0.      , 0.000684, 0.001369, ..., 0.149213, 0.149897, 0.150582])
    • " ], "text/plain": [ "\n", diff --git a/examples/symba_swifter_comparison/mars_disk/param.swifter.in b/examples/symba_swifter_comparison/mars_disk/param.swifter.in index f5cbe9a08..4bdc100aa 100644 --- a/examples/symba_swifter_comparison/mars_disk/param.swifter.in +++ b/examples/symba_swifter_comparison/mars_disk/param.swifter.in @@ -1,5 +1,5 @@ T0 0.0 -TSTOP 6000.0 +TSTOP 3000.0 DT 600.0 PL_IN pl.swifter.in TP_IN tp.in diff --git a/examples/symba_swifter_comparison/mars_disk/param.swiftest.in b/examples/symba_swifter_comparison/mars_disk/param.swiftest.in index 406344ff0..0d48de602 100644 --- a/examples/symba_swifter_comparison/mars_disk/param.swiftest.in +++ b/examples/symba_swifter_comparison/mars_disk/param.swiftest.in @@ -1,6 +1,6 @@ !Parameter file for the SyMBA-RINGMOONS test T0 0.0 -TSTOP 6000.0 +TSTOP 3000.0 DT 600.0 PL_IN pl.swiftest.in TP_IN tp.in diff --git a/examples/symba_swifter_comparison/mars_disk/swiftest_vs_swifter.ipynb b/examples/symba_swifter_comparison/mars_disk/swiftest_vs_swifter.ipynb new file mode 100644 index 000000000..960e37a53 --- /dev/null +++ b/examples/symba_swifter_comparison/mars_disk/swiftest_vs_swifter.ipynb @@ -0,0 +1,557 @@ +{ + "cells": [ + { + "cell_type": "code", + "execution_count": 1, + "metadata": {}, + "outputs": [], + "source": [ + "import numpy as np\n", + "import swiftest\n", + "import matplotlib.pyplot as plt" + ] + }, + { + "cell_type": "code", + "execution_count": 2, + "metadata": {}, + "outputs": [ + { + "name": "stdout", + "output_type": "stream", + "text": [ + "Reading Swifter file param.swifter.in\n", + "Reading in time 2.520e+03\n", + "Creating Dataset\n", + "Successfully converted 43 output frames.\n", + "Swifter simulation data stored as xarray DataSet .ds\n" + ] + } + ], + "source": [ + "swiftersim = swiftest.Simulation(param_file=\"param.swifter.in\", codename=\"Swifter\")\n", + "swiftersim.bin2xr()" + ] + }, + { + "cell_type": "code", + "execution_count": 3, + "metadata": {}, + "outputs": [ + { + "name": "stdout", + "output_type": "stream", + "text": [ + "Reading Swiftest file param.swiftest.in\n", + "Reading in time 2.520e+03\n", + "Creating Dataset\n", + "Successfully converted 43 output frames.\n", + "\n", + "Adding particle info to Dataset\n", + "Swiftest simulation data stored as xarray DataSet .ds\n" + ] + } + ], + "source": [ + "swiftestsim = swiftest.Simulation(param_file=\"param.swiftest.in\")\n", + "swiftestsim.bin2xr()" + ] + }, + { + "cell_type": "code", + "execution_count": 4, + "metadata": {}, + "outputs": [], + "source": [ + "swiftdiff = swiftestsim.ds - swiftersim.ds" + ] + }, + { + "cell_type": "code", + "execution_count": 5, + "metadata": {}, + "outputs": [ + { + "name": "stderr", + "output_type": "stream", + "text": [ + "No handles with labels found to put in legend.\n" + ] + }, + { + "data": { + "image/png": "iVBORw0KGgoAAAANSUhEUgAAAYIAAAERCAYAAAB2CKBkAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjMuNCwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy8QVMy6AAAACXBIWXMAAAsTAAALEwEAmpwYAAAYtElEQVR4nO3dfZDlVZ3f8fe3H2bGeXC6hxlgwBGYkcQHSgEniDqoa9wI1Ca41poC19U1KOiqtWzF1JLaql0rW1ZMLKiERWXZlWVJbSRZlxhixkXjGgUzKoPyzBJnGgwNg9PM89O9fbv7mz/u7aEZeoYB+vT9df/er6quvg+//p1z+jL94Zzz+50TmYkkqb56ul0BSVJ3GQSSVHMGgSTVnEEgSTVnEEhSzRkEklRzczIIIuKmiNgeEQ/O0PleHRHfjohHIuLhiDh9Js4rSXPBnAwC4Gbgwhk83y3AFzPzdcB5wPYZPLckVdqcDILM/AGwc+prEbEuIv42Iu6JiDsj4rXHc66IeD3Ql5nf6Zx7f2YenPlaS1I1zckgOIobgc9k5puBzwJfPs6f+wfA7oi4LSJ+FhFfjIjeYrWUpIrp63YFZkJELAXeBvx1REy+vLDz3vuBfzPNjz2Zme+l/Tu4ADgH+H/AfwF+G/hq2VpLUjXMiyCg3bPZnZlnH/lGZt4G3HaMnx0GfpaZQwAR8Q3gfAwCSTUxL4aGMnMv8FhEfAAg2t50nD9+NzAYEas6z98NPFygmpJUSXMyCCLia8Am4B9GxHBEXA78JnB5RNwHPARccjznysxx2nMK342IB4AA/qxMzSWpesJlqCWp3uZkj0CSNHPm3GTxypUr8/TTT+92NSRpTrnnnnueycxV070354Lg9NNPZ/Pmzd2uhiTNKRHxi6O959CQJNVcsSCIiDUR8b3OQm4PRcTvTnNMRMR1EbElIu6PiHNL1UeSNL2SQ0NjwL/MzJ9GxDLgnoj4TmZOvUb/IuDMztdbgK90vkuSZkmxHkFmbsvMn3Ye7wMeAU494rBLgFuy7UfAQESsLlUnSdLzzcocQWd9/3OAHx/x1qnAE1OeD/P8sCAiroiIzRGxeWRkpFg9JamOigdBZ0G4vwGu6iwF8Zy3p/mR593hlpk3Zub6zFy/atW0Vz9Jkl6iokEQEf20Q+CvOou/HWkYWDPl+auAp0rWSZL0XCWvGgraK3g+kpnXHuWw24EPd64eOh/Yk5nbStVJkuaqTV//Go/f/7Mi5y7ZI3g78FvAuyPi3s7XxRHxiYj4ROeYjcAQsIX2Qm+/U7A+kjRn/ei2Wxl++IEi5y52+Whm3sX0cwBTj0ngU6XqIEnzwfhYi4nxcfoXLipyfu8slqSKazWaAPQvXFjk/AaBJFVcq9kAoM8gkKR6mgwCh4YkqaZaDYNAkmrNHoEk1Vyr2ZksXuQcgSTVkj0CSao55wgkqeaeHRoyCCSplp4dGnKOQJJqaXJoyBvKJKmmxkab9PUvoKent8j5DQJJqrhWs0FfofkBMAgkqfJajWax+QEwCCSp8lrNRrFLR8EgkKTKaweBPQJJqi17BJJUc84RSFLN2SOQpJprNRvFlpcAg0CSKq/VbNJnj0CS6mus4VVDklRbExPjjLVGnSOQpLoaK7wENRgEklRph/cisEcgSfX0bBA4RyBJtXR4UxqHhiSpnkrvVwwGgSRV2uEewQKHhiSplkpvXA8GgSRV2mSPoNR+xWAQSFKlOUcgSTU35lVDklRvc/qGsoi4KSK2R8SDR3n/XRGxJyLu7Xz9Yam6SNJc1Wo2iJ4eevv6ipVR7sxwM3A9cMsxjrkzM3+tYB0kaU5rNdqb0kREsTKK9Qgy8wfAzlLnl6Q6KL0pDXR/juCtEXFfRHwrIt5wtIMi4oqI2BwRm0dGRmazfpLUVa1m2f2KobtB8FPgtMx8E/AnwDeOdmBm3piZ6zNz/apVq2arfpLUdaX3K4YuBkFm7s3M/Z3HG4H+iFjZrfpIUhVNzhGU1LUgiIiTozP7ERHndeqyo1v1kaQqao02i88RFLtqKCK+BrwLWBkRw8AfAf0AmXkD8BvAJyNiDDgEXJqZWao+kjQXjTUaLFk+ULSMYkGQmZe9wPvX0768VJJ0FK1mk76CK49C968akiQdQx0uH5UkHcO8vmpIknRsmUmr0TQIJKmuxlstMifm9Q1lkqRjmI2N68EgkKTKOhwEDg1JUj21GpN7ETg0JEm15NCQJNWcQ0OSVHMGgSTVnHMEklRzzhFIUs2NNSd7BAaBJNXSZI+gz6EhSaqnw0NDLkMtSfU0uRdB9JT9U20QSFJFtfcrLtsbAINAkiprNjalAYNAkiprNjalAYNAkiqr1Ww6NCRJddaeI7BHIEm15RyBJNVcq9mkzx6BJNVXe7LYOQJJqq0x5wgkqd6cI5CkGpsYH2d8bMyhIUmqq9bkEtSFF5wDg0CSKmm2NqUBg0CSKmm29isGg0CSKqnVMAgkqdYOzxE4WSxJ9XR4m0rnCCSpnpwjkKSaG5sPcwQRcVNEbI+IB4/yfkTEdRGxJSLuj4hzS9VFkuaaw3MEi+b2HMHNwIXHeP8i4MzO1xXAVwrWRZLmlHkxNJSZPwB2HuOQS4Bbsu1HwEBErC5VH0maS+py+eipwBNTng93XnueiLgiIjZHxOaRkZFZqZwkdVOr2aCnt4/evr7iZXUzCGKa13K6AzPzxsxcn5nrV61aVbhaktR9rWZzVuYHoLtBMAysmfL8VcBTXaqLJFVKq9mYlQXnoLtBcDvw4c7VQ+cDezJzWxfrI0mV0e4RlJ8fACg2+BQRXwPeBayMiGHgj4B+gMy8AdgIXAxsAQ4CHy1VF0maa1rNxqzsVwwFgyAzL3uB9xP4VKnyJWkuG2vOzjaV4J3FklRJrUZzVhacA4NAkiqpZY9AkupttjauB4NAkiqp1XRoSJJqrdVwaEiSaiszaY3O3n0EBoEkVczYaBMyn9MjGB9rFSvvBYMgIr4aEWcf8drnSlVIkupuuv2Kv/yxD3Lnf765SHnH0yN4L3BzRHx4ymv/rEhtJEnPW4J69NBBRg8dYuGSpUXKO54g2A68A/hARHwpIvqYfuVQSdIMOLwpTWeOYP+uXQAsHVxRpLzjmiPIzL2Z+U+BEeD7wPIitZEkHQ6Cvs7qowd2t/f4WtLFIHhNRJwKkJmfA/4t8HiR2kiSaDUm5wjaPYIDu9pBUKpHcDyLzl0LfDsidgK3Al/PzHcXqY0kqX3VEM9uXL+/EwRLBrrUI8jMz2XmG2ivFHoK8P2I+F9FaiNJet7G9Qd276KvfwELlywpUt6LuY9gO/A0sAM4sUhtJEnPu2rowK6dLBkcJKLMdTrHcx/BJyPifwPfBVYCH8/MNxapjSRpmquGdhYbFoLjmyM4DbgqM+8tVgtJ0mFH3lB2YNdOVq45rVh5xzNHcLUhIEmzp9VsQMSUy0d3Fbt0FFxrSJIqp9Vo0L9gIRFBq9mgefAASwYGi5VnEEhSxUzdlObA5F3FK04oVp5BIEkVM3VTmv2TdxXbI5Ck+pi6Kc3kXcXOEUhSjUzduL708hJgEEhS5bSazWeXl9i9i96+PhYtXVasPINAkiqm1Ww8e+nozh0sHih3VzEYBJJUOWNThob2795VdFgIDAJJqpxWY+rlo2WXlwCDQJIqpzXaPGLBOYNAkmql1WjfRzA2OkrjwH6HhiSpTsbHxpgYH6N/4SIO7G7fVbxksNzNZGAQSFKlTF2CenJnsqXOEUhSfUzdnaz0pvWTDAJJqpBnN65fOCt3FYNBIEmVMtkj6OsMDfX09vKKZa8sWmbRIIiICyPi0YjYEhFXT/P+uyJiT0Tc2/n6w5L1kaSqe87Q0K5d7buKe8r+P/vxbFX5kkREL/Al4FeBYeDuiLg9Mx8+4tA7M/PXStVDkuaSscZz5wiWFlx+elLJmDkP2JKZQ5k5CtwKXFKwPEma86buV7x/Fm4mg7JBcCrwxJTnw53XjvTWiLgvIr4VEW+Y7kQRcUVEbI6IzSMjIyXqKkmVMPXy0QO7dhafKIayQTDdUnl5xPOfAqdl5puAPwG+Md2JMvPGzFyfmetXrVo1s7WUpAqZDIKenl4O7dtbfJ0hKBsEw8CaKc9fBTw19YDM3JuZ+zuPNwL9EbGyYJ0kqdJanTmCyUAofVcxlA2Cu4EzI+KMiFgAXArcPvWAiDg5OotsR8R5nfrsKFgnSaq0yTmC5sEDACwdLLdp/aRiVw1l5lhEfBq4A+gFbsrMhyLiE533bwB+A/hkRIwBh4BLM/PI4SNJqo1Ws0FvXx8H9+4Bym5aP6lYEMDh4Z6NR7x2w5TH1wPXl6yDJM0lrWZ7CeoDO2dneQnwzmJJqpRWs0HfovY9BBE9LF6+vHiZBoEkVchkj2D/rl0sHhigp6e3eJkGgSRVSHu/4oUc2LVjVuYHwCCQpEppNdob18/GpvWTDAJJqpBWs3H4ruLZmCgGg0CSKqXVbNK3YAEH9+6ZlbuKwSCQpEppNRtE9ECmQ0OSVEetRgM699U6NCRJNdRqNsmcAJiVvQjAIJCkysiJCcZGm4yPjQOwZIU9AkmqldZoe8G58dYoRLBkebtHkJnccMMNbNq0qUi5BoEkVcTkEtRjo6MsfuVyenrbdxXv2LGDp59+mv7+/iLlGgSSVBGTS1C3mo3nTBRv3boVgLVr1xYp1yCQpIoY62xGM3ro4HMmioeGhhgYGGBFoTkDg0CSKmLqpjSTPYLx8XEef/zxYr0BMAgkqTImt6dsHjhw+Gayp556imazybp164qVaxBIUkVMBkFmHl5eYnJ+4IwzzihWrkEgSRUxedUQPLtp/dDQEKtXr2bx4sXFyjUIJKkiJucIoL1pfbPZZHh4uOiwEBgEklQZk0ND0O4R/OIXv2BiYqLoRDEYBJJUGc8ZGhoYZOvWrfT19bFmzZqi5RoEklQRk0NDr1j2Snr7+hkaGuLVr351sTuKJxkEklQRrWaD6OlhyeAK9u7dy8jISPH5ATAIJKky2pvSBEsGBnnssceAcstKTGUQSFJFtBoNMpOlgyewdetWFi9ezEknnVS8XINAkipitNEgJyZYMjDI0NAQa9eupaen/Wf6loduYevurUXKNQgkqSKaB/cDkP397N+///Cw0D2/vIcvbv4it2+9vUi5BoEkVUTz4AEA9h5qXz20du1aWuMt/njTH3PKklO48o1XFim3r8hZJUkv2uihQwCM7NnDihUrGBgY4M8f+HO27tnK9e++nsX9ZZaZsEcgSRUxeUPZtmd2sG7dOob3DfOn9/0p73n1e3jnmncWK9cgkKSKGBsdBWA0g7Vr1/L5H3+enujh98/7/aLlGgSSVBHjrRbR20f09jLUM8RdT97Fp8/5NCcvOblouQaBJFVAZjIxPkb293PyKSdzzX3X8LoVr+Oy115WvGwniyWpAsbHxkhgvLeP7Yu288yhZ7ju3dfR11P+z7Q9AkmqgFazARFk3wL+7sDfcelrL+WslWfNStlFgyAiLoyIRyNiS0RcPc37ERHXdd6/PyLOLVkfSaqq0UOHIJOJvl56Bnv4zDmfed4xE2NjRcou1ueIiF7gS8CvAsPA3RFxe2Y+POWwi4AzO19vAb7S+S5JtbJ/1w4CONTb5DOnfYif/+3X2Xn/Zg4NbWViZIQ4cIixs1/L+6+9bcbLLjn4dB6wJTOHACLiVuASYGoQXALckpkJ/CgiBiJidWZum+nK/PNvfImnF58406eVpJlz+UcA+P5eYBFw3rvaXx2nHBzh/QWKLTk0dCrwxJTnw53XXuwxRMQVEbE5IjaPjIzMeEUlqc5K9ghimtfyJRxDZt4I3Aiwfv36571/PP7r+z71Un5MkmbNW+7YxODjQ7zvwHae3ruH5cuXc8EFF3D22WfT11fuz3XJHsEwMHWjzVcBT72EYySpFt65ZjWPnraOX7nrLj542WUsXbqUb37zm1x33XX85Cc/odVqFSm3ZBDcDZwZEWdExALgUuDINVRvBz7cuXrofGBPifkBSZoLNqwc4NDCRTxwqMWJDzzAxz72MT70oQ+xfPlyNm7cyB133FGk3GJ9jcwci4hPA3cAvcBNmflQRHyi8/4NwEbgYmALcBD4aKn6SFLVvW1gKQAP/Mp7eON/+I8su/AiXvOa17Bu3Toee+wxli9fXqTcaF+wM3esX78+N2/e3O1qSFIR//juv2d5o8HnP34ZJ1x5JSf+3lUzct6IuCcz10/3nncWS1KFbBhYxj0TwYJL3sfOv/gLWk8+WbxMg0CSKmTD4FKaE8kTH/8E9PSw/Zpri5dpEEhShZw/sJTegE09CzjhX3yUvRs3cvBnPytapkEgSRWyrK+Xc5Yt5q5d+zjh8svpW7WKX37hC5SczzUIJKliNgwu4959BzmwcBGrrrqKxn33s/d/bixWnkEgSRWzYXAp4wmbdu9n+a+/j4Wvfx3br7mGic6exjPNIJCkiln/yiUs6gl+uGs/0dPDSVdfzdi2bey8+eYi5RkEklQxi3p7+EfLl3Dnrn0ALDnvPE648koWr5/2NoCXzSCQpAraMLCMhw80eGa0vRnNib93lUEgSXWyYbC93MQPd+8rXpZBIEkV9KZli1na28MPd+0vXpZBIEkV1NcTvHVgKXcZBJJUXxcMLmXoUJMnG6NFyzEIJKmiNgwuAyjeKzAIJKmiXrtkESf09x2+jLQUg0CSKqongrcPLuWHu/e71pAk1dUFg0vZ1mwxdKhZrAyDQJIqbMNAe57gzoLzBAaBJFXY6a9YwKkL+7mr4DyBQSBJFRYRbBhcxv/ZvZ+JQvMEBoEkVdyGwaXsbI3z8P5DRc5vEEhSxU2uO1TqfgKDQJIqbvXCBbz/pEFOXNhf5Px9Rc4qSZpRX379acXObY9AkmrOIJCkmjMIJKnmDAJJqjmDQJJqziCQpJozCCSp5gwCSaq5KLnZQQkRMQL84iX++ErgmRmszlxQtzbXrb1Qvzbb3pfmtMxcNd0bcy4IXo6I2JyZ67tdj9lUtzbXrb1Qvzbb3pnn0JAk1ZxBIEk1V7cguLHbFeiCurW5bu2F+rXZ9s6wWs0RSJKer249AknSEQwCSaq52gRBRFwYEY9GxJaIuLrb9ZkpEfF4RDwQEfdGxObOaysi4jsR8fPO98Epx//rzu/g0Yh4b/dqfvwi4qaI2B4RD0557UW3MSLe3PldbYmI6yIiZrstx+Mo7f1cRDzZ+ZzvjYiLp7w319u7JiK+FxGPRMRDEfG7ndfn5Wd8jPZ27zPOzHn/BfQCW4G1wALgPuD13a7XDLXtcWDlEa/9e+DqzuOrgX/Xefz6TtsXAmd0fie93W7DcbTxHcC5wIMvp43AT4C3AgF8C7io2217Ee39HPDZaY6dD+1dDZzbebwM+L+dds3Lz/gY7e3aZ1yXHsF5wJbMHMrMUeBW4JIu16mkS4C/7Dz+S+B9U16/NTObmfkYsIX276bSMvMHwM4jXn5RbYyI1cArM3NTtv8F3TLlZyrlKO09mvnQ3m2Z+dPO433AI8CpzNPP+BjtPZri7a1LEJwKPDHl+TDH/sXPJQl8OyLuiYgrOq+dlJnboP0fHXBi5/X59Ht4sW08tfP4yNfnkk9HxP2doaPJYZJ51d6IOB04B/gxNfiMj2gvdOkzrksQTDduNl+um317Zp4LXAR8KiLecYxj5/PvYdLR2jjX2/4VYB1wNrANuKbz+rxpb0QsBf4GuCoz9x7r0Glem3Ntnqa9XfuM6xIEw8CaKc9fBTzVpbrMqMx8qvN9O/DfaA/1/LLTbaTzfXvn8Pn0e3ixbRzuPD7y9TkhM3+ZmeOZOQH8Gc8O6c2L9kZEP+0/in+Vmbd1Xp63n/F07e3mZ1yXILgbODMizoiIBcClwO1drtPLFhFLImLZ5GPgnwAP0m7bRzqHfQT4753HtwOXRsTCiDgDOJP2ZNNc9KLa2Bla2BcR53eurPjwlJ+pvMk/iB2/TvtzhnnQ3k79vgo8kpnXTnlrXn7GR2tvVz/jbs+gz9YXcDHt2fmtwB90uz4z1Ka1tK8muA94aLJdwAnAd4Gfd76vmPIzf9D5HTxKBa+oOEo7v0a7q9yi/X9Bl7+UNgLrO/+4tgLX07mzvmpfR2nvfwIeAO7v/GFYPY/au4H2kMb9wL2dr4vn62d8jPZ27TN2iQlJqrm6DA1Jko7CIJCkmjMIJKnmDAJJqjmDQJJqziCQXkBEDETE73QenxIRX+92naSZ5OWj0gvorAfzzcw8q9t1kUro63YFpDngC8C6iLiX9s1Nr8vMsyLit2mv9tgLnEV7bZgFwG8BTeDizNwZEeuALwGrgIPAxzPz72e7EdLRODQkvbCrga2ZeTbwr4547yzgg7TXhfk8cDAzzwE20b7lH9qbj38mM98MfBb48mxUWjpe9gikl+d72V5Tfl9E7AH+R+f1B4A3dlaYfBvw11M2j1o4+9WUjs4gkF6e5pTHE1OeT9D+99UD7O70JqRKcmhIemH7aG8p+KJle535xyLiA9BeeTIi3jSTlZNeLoNAegGZuQP4YbQ3k//iSzjFbwKXR8TkKrHzeZtUzUFePipJNWePQJJqziCQpJozCCSp5gwCSao5g0CSas4gkKSaMwgkqeb+P+sWuGyVH0YiAAAAAElFTkSuQmCC\n", + "text/plain": [ + "
    " + ] + }, + "metadata": { + "needs_background": "light" + }, + "output_type": "display_data" + }, + { + "name": "stdout", + "output_type": "stream", + "text": [ + "\n" + ] + } + ], + "source": [ + "fig, ax = plt.subplots()\n", + "swiftdiff['vz'].plot.line(ax=ax, x=\"time\")\n", + "legend = ax.legend()\n", + "legend.remove()\n", + "plt.show()\n", + "print(\"\")" + ] + }, + { + "cell_type": "code", + "execution_count": 9, + "metadata": {}, + "outputs": [], + "source": [ + "last = swiftdiff.isel(time=-2)" + ] + }, + { + "cell_type": "code", + "execution_count": 10, + "metadata": {}, + "outputs": [], + "source": [ + "badval = last.where(last['pz'] != 0, drop=True)" + ] + }, + { + "cell_type": "code", + "execution_count": 11, + "metadata": {}, + "outputs": [ + { + "data": { + "text/html": [ + "
    \n", + "\n", + "\n", + "\n", + "\n", + "\n", + "\n", + "\n", + "\n", + "\n", + "\n", + "\n", + "\n", + "\n", + "\n", + "
    <xarray.Dataset>\n",
+       "Dimensions:  (id: 1)\n",
+       "Coordinates:\n",
+       "  * id       (id) float64 85.0\n",
+       "    time     float64 2.46e+03\n",
+       "Data variables:\n",
+       "    GMass    (id) float64 0.0\n",
+       "    Radius   (id) float64 0.0\n",
+       "    px       (id) float64 0.0\n",
+       "    py       (id) float64 0.0\n",
+       "    pz       (id) float64 -1.819e-12\n",
+       "    vx       (id) float64 0.0\n",
+       "    vy       (id) float64 0.0\n",
+       "    vz       (id) float64 -1.11e-16
    " + ], + "text/plain": [ + "\n", + "Dimensions: (id: 1)\n", + "Coordinates:\n", + " * id (id) float64 85.0\n", + " time float64 2.46e+03\n", + "Data variables:\n", + " GMass (id) float64 0.0\n", + " Radius (id) float64 0.0\n", + " px (id) float64 0.0\n", + " py (id) float64 0.0\n", + " pz (id) float64 -1.819e-12\n", + " vx (id) float64 0.0\n", + " vy (id) float64 0.0\n", + " vz (id) float64 -1.11e-16" + ] + }, + "execution_count": 11, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "badval" + ] + }, + { + "cell_type": "code", + "execution_count": null, + "metadata": {}, + "outputs": [], + "source": [] + } + ], + "metadata": { + "kernelspec": { + "display_name": "swiftestOOF", + "language": "python", + "name": "swiftestoof" + }, + "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.7.10" + } + }, + "nbformat": 4, + "nbformat_minor": 4 +}