|
|
{ "cells": [ { "cell_type": "code", "execution_count": 42, "metadata": {}, "outputs": [ { "data": { "image/png": "iVBORw0KGgoAAAANSUhEUgAAAiwAAAGdCAYAAAAxCSikAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjUuMSwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy/YYfK9AAAACXBIWXMAAA9hAAAPYQGoP6dpAABFTUlEQVR4nO3deVxVdeLG8c9ldwPXEFNxyX1FVBYFqQzTNpsaqSmzps2y1GjTzCWnX2SlpaWmZpplRIVbM1raFKKCG4HmrmniAhkuXHBhPb8/KCZy4xJw7oXn/XrdPzh+7/G5J/M+fs8532MxDMNARERExI45mR1ARERE5GpUWERERMTuqbCIiIiI3VNhEREREbunwiIiIiJ2T4VFRERE7J4Ki4iIiNg9FRYRERGxey5mBygvhYWFHD9+nDp16mCxWMyOIyIiIqVgGAZZWVk0adIEJ6fLz6NUmcJy/PhxmjVrZnYMERERKYMjR47QtGnTy/56lSksderUAYo+sKenp8lpREREpDSsVivNmjUr/h6/nCpTWH4/DeTp6anCIiIi4mCudjmHLroVERERu6fCIiIiInZPhUVERETsngqLiIiI2D0VFhEREbF7KiwiIiJi91RYRERExO6psIiIiIjdU2ERERERu6fCIiIiInZPhUVERETsngqLiIiI2L0q8/BDERERqRhRUZCdDU89BT4+5mRQYREREZHLys6GN96AM2cgIABuv92cHDolJCIiIpe1cGFRWbmu/klu7XvGtBwqLCIiInJJBQXwzrRCAJ45NR6nNd+YlkWFRURERC7pq6/gp0NO1OMUw9puhLvvNi2LCouIiIhc0rQ3CwAYzvvUmvAsODublkUX3YqIiMhFtmyBdQnOuJLLU61WwT1xpubRDIuIiIhc5O038wG4h89oMvFRU2dXQDMsIiIi8idHjsDnsUVzGs9c+wX8Y6nJiTTDIiIiIn/y7rQ8CgqduJ7v8Jt8J7iYP7+hwiIiIiLFsrJg7vtFtzJHXrMYhg41OVERFRYREREptmBuHpkX3GnLXgZNDgRXV7MjASosIiIi8puCAngn6hwAz9T7CKeHhpmc6H9UWERERASA5V/mceikF/U5yQMTW4Kbm9mRiqmwiIiICADTxmUA8ESdT6g5/AGT05SkwiIiIiJsWp/Hhp98cCWXES/UBnd3syOVoMIiIiIivB2ZCsA/PJbi8+w/TE5zMRUWERGRau7wgTy+3OILwDNP5UGNGiYnupgKi4iISDX37lN7KcCFG13j6fbK38yOc0kqLCIiItWY9VQ+89YUza5EDv0VatY0OdGlqbCIiIhUYx8+nYy1sA7tnfdx89sDzI5zWSosIiIi1dTZkxeYGtMUgGdu/wknz9omJ7o8FRYREZFq6l+Dkzha4IOv8xGGvt/X7DhXpMIiIiJSDe36OpWp63sD8G7kz9S4po7Jia5MhUVERKSaMQoNRtx3mnxcua1hIrdNse/ZFVBhERERqXY+jdxK3Klu1OAc0z/zBovF7EhXpcIiIiJSjZw5ms2z7xbdxvzy9Qm0vLGVyYlKx+bCEh8fz2233UaTJk2wWCwsW7bsiuPj4uKwWCwXvfbs2VNiXGxsLB07dsTd3Z2OHTuydOlSW6OJiIjIVUy4PYVfCq+hretBnl3Sx+w4pWZzYTl79izdunXjvffes+l9e/fuJS0trfjVpk2b4l9LTEwkIiKCoUOHsm3bNoYOHcqQIUPYtGmTrfFERETkMn74/AAzk4MAmDnxV9zr2t8S/JdjMQzDKPObLRaWLl3K4MGDLzsmLi6O66+/ntOnT1O3bt1LjomIiMBqtbJq1aribTfffDP16tUjOjq6VFmsViteXl5kZmbi6elpy8cQERGp8goLDILr7WJTVicirl3PZ0ft40Lb0n5/V9o1LH5+fvj4+HDjjTfy/fffl/i1xMREwsPDS2wbMGAACQkJl91fTk4OVqu1xEtEREQubf4jiWzK6kQdrExb2tLsODar8MLi4+PD3LlziY2NZcmSJbRr144bb7yR+Pj44jHp6el4e3uXeJ+3tzfp6emX3W9UVBReXl7Fr2bNmlXYZxAREXFkGftPM+aj9gBMvm0rTXpda3Ii27lU9G/Qrl072rVrV/xzUFAQR44c4a233iI0NLR4u+VPt1QZhnHRtj8aO3YskZGRxT9brVaVFhERkUsYc/tOThl96eq+l6diQsyOUyam3NYcGBjI/v37i39u3LjxRbMpJ06cuGjW5Y/c3d3x9PQs8RIREZGSEj/Yyfw9RderzJp6HpcariYnKhtTCktycjI+Pj7FPwcFBbFmzZoSY1avXk1wcHBlRxMREaky8nMKeGJk0cmUh66Lp8+I7uYG+gtsPiWUnZ3NgQMHin8+dOgQKSkp1K9fn+bNmzN27FiOHTvGokWLAHjnnXdo0aIFnTp1Ijc3l08++YTY2FhiY2OL9zFq1ChCQ0OZMmUKd9xxB8uXL+fbb79l/fr15fARRUREqqdZ921g2/lQ6llOM2V5B7Pj/CU2F5atW7dy/fXXF//8+3Ukw4YNY+HChaSlpZGamlr867m5uTz33HMcO3aMGjVq0KlTJ/7zn/8waNCg4jHBwcF89tlnvPzyy4wfP57WrVsTExNDQEDAX/lsIiIi1dZP36cyLtYPgNfv3U6jjv1MTvTX/KV1WOyJ1mEREREpkp9TQGijXSRmdSHUM4XvM7rg5OpsdqxLsrt1WERERKRyTLl1HYlZXfAkk0X/aWC3ZcUWKiwiIiJVyNaPdzPp26JnBL332I/49q0aS36osIiIiFQR5zLOcf8j7uTjyt+bJnD/bMd5uOHVqLCIiIhUES/csJW9ua3wcUrn/e/bY3G6/AKsjkaFRUREpAr4+v+SmPlj0QryC189Rv3r6pucqHypsIiIiDi4jH2neGhCUwCe7rqW8LH+JicqfyosIiIiDswoNHj8+n2kF3rTwe0AU77rZXakCqHCIiIi4sAWPb6BJccDcSGPTz7Mo0aDmmZHqhAqLCIiIg7qUPwRnv6gKwCTw9fT4z7HXn7/SlRYREREHFBBbgEP3HaKLDzpU2c7L3wVanakCqXCIiIi4oDevH0d663dqE0WH/+7Hs5ujr+a7ZWosIiIiDiY9bO2M/6bokXhZjyUQsvQqrGa7ZWosIiIiDiQX3b8ypCnryEfV+713cCDH/Q1O1KlUGERERFxEPkX8rk35ChphY3p4PYTczd2q1Kr2V6JCouIiIiDmHDDer4/40ctsomNhdqNa5sdqdKosIiIiDiAFeM2EZUYBsD8kdvpcGtrcwNVMhUWERERO3cwLpUHXmsPwMhua4mYHmxyosqnwiIiImLHzp86z12DzpGJF0G1f+TN9UFmRzKFCouIiIgdezpoKynn29PQksHn3zXErbab2ZFMocIiIiJipz58aB3z94VgoZDo11Np2svH7EimUWERERGxQykxexmxsCcAk2+Mp/8LPUxOZC4VFhERETtz5nAmd93vwQVqMKjRFl76umo/J6g0VFhERETsSGF+IcMC93Aw35cWLkf4OPE6nFz0da0jICIiYkdeDo1nRXoAbuTwxfws6reuZ3Yku6DCIiIiYic+Hr7+f4vDDd9Kzwc6mhvIjqiwiIiI2IHEuT/yyJxeAIwNiuP+2X1MTmRfVFhERERMdnjDUQYPb0wu7gz22cir8brI9s9UWEREREyUnZ7N7f3PcsJoRDePvXz8Q2ddZHsJOiIiIiImKcwv5H6/nWy/0A5vpxOs+K52tXoCsy1UWEREREwyLiSe5ekBuHOBZXNO0DzoWrMj2S0VFhERERN8PHw9r28MA2D+E0kEPtLZ3EB2ToVFRESkkiXM+d8dQS8Fx3HfLN0RdDUqLCIiIpXo8Iaj3PlE0R1Bd/ps5F9rdUdQaaiwiIiIVJLM1Exu63+OE0YjutfYw8cpXXRHUCnpKImIiFSCHGsOd3Y7yI8X2tLY6RdWfO9JrWtqmR3LYaiwiIiIVLDC/EIe6rqV78/4UZssVn5ymmYBTcyO5VBUWERERCrYmOB4og/3wYU8lkTtw+/e9mZHcjgqLCIiIhVo+t/W8uaWMAA "text/plain": [ "<Figure size 640x480 with 1 Axes>" ] }, "metadata": {}, "output_type": "display_data" } ], "source": [ "import numpy as np\n", "import mpmath as mp\n", "import matplotlib.pyplot as plt\n", "\n", "def polylog(power, numerator):\n", " \n", " order = 100\n", " \n", " dataShape = numerator.shape\n", " numerator = np.tile(numerator, (order, 1))\n", " numerator = np.power(numerator.T, np.arange(1, order+1)).T\n", "\n", " denominator = np.arange(1, order+1)\n", " denominator = np.tile(denominator, (dataShape[0], 1))\n", " denominator = denominator.T\n", "\n", " data = numerator/ np.power(denominator, power)\n", "\n", " return np.sum(data, axis=0)\n", "\n", "x = np.linspace(0, 1, 51)\n", "y1 = polylog(2, x)\n", "y2 = [float(mp.polylog(2, i).real) for i in x]\n", "\n", "plt.figure()\n", "\n", "plt.plot(x, y1, 'r')\n", "plt.plot(x, y2, 'b')\n", "\n", "plt.show()" ] }, { "cell_type": "code", "execution_count": 104, "metadata": {}, "outputs": [], "source": [ "from lmfit.lineshapes import (not_zero, breit_wigner, damped_oscillator, dho, doniach,\n", " expgaussian, exponential, gaussian, gaussian2d,\n", " linear, lognormal, lorentzian, moffat, parabolic,\n", " pearson7, powerlaw, pvoigt, rectangle, sine,\n", " skewed_gaussian, skewed_voigt, split_lorentzian, step,\n", " students_t, thermal_distribution, tiny, voigt)\n", "\n", "def polylog(power, numerator):\n", " \n", " order = 100\n", " \n", " dataShape = numerator.shape\n", " numerator = np.tile(numerator, (order, 1))\n", " numerator = np.power(numerator.T, np.arange(1, order+1)).T\n", "\n", " denominator = np.arange(1, order+1)\n", " denominator = np.tile(denominator, (dataShape[0], 1))\n", " denominator = denominator.T\n", "\n", " data = numerator/ np.power(denominator, power)\n", "\n", " return np.sum(data, axis=0)\n", "\n", "def polylog2_2d(x, y=0.0, centerx=0.0, centery=0.0, amplitude=1.0, sigmax=1.0, sigmay=1.0): \n", " ## Approximation of the polylog function with 2D gaussian as argument. -> discribes the thermal part of the cloud\n", " return amplitude / 2 / 5.403642092095097 / max(tiny, sigmax * sigmay) * polylog(2, np.exp( -((x-centerx)**2/(2 * (sigmax)**2))-((y-centery)**2/( 2 * (sigmay)**2)) ))\n" ] }, { "cell_type": "code", "execution_count": 95, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "5.403642092095097\n" ] } ], "source": [ "from scipy import special\n", "\n", "sum = 0\n", "for i in range(1,20000):\n", " sum += 1/i**4 * special.gamma(1/2/i)**2\n", " \n", "print(sum)" ] }, { "cell_type": "code", "execution_count": 98, "metadata": {}, "outputs": [ { "data": { "text/plain": [ "4.0" ] }, "execution_count": 98, "metadata": {}, "output_type": "execute_result" } ], "source": [ "x[1] - x[0] " ] }, { "cell_type": "code", "execution_count": 105, "metadata": {}, "outputs": [ { "data": { "text/plain": [ "0.8405962721688879" ] }, "execution_count": 105, "metadata": {}, "output_type": "execute_result" } ], "source": [ "x = np.linspace(-100, 100, 101)\n", "y = np.linspace(-100, 100, 101)\n", "\n", "X, Y = np.meshgrid(x, y)\n", "X = X.flatten()\n", "Y = Y.flatten()\n", "Z = polylog2_2d(x=X, y=Y).reshape(101, 101)\n", "\n", "np.sum(Z)" ] }, { "attachments": {}, "cell_type": "markdown", "metadata": {}, "source": [ "# Import supporting package" ] }, { "cell_type": "code", "execution_count": 43, "metadata": {}, "outputs": [], "source": [ "import xarray as xr\n", "import pandas as pd\n", "import numpy as np\n", "import copy\n", "\n", "import glob\n", "\n", "import xrft\n", "import finufft\n", "\n", "from uncertainties import ufloat\n", "from uncertainties import unumpy as unp\n", "from uncertainties import umath\n", "\n", "from datetime import datetime\n", "\n", "import matplotlib.pyplot as plt\n", "plt.rcParams['font.size'] = 18\n", "\n", "from DataContainer.ReadData import read_hdf5_file, read_hdf5_global, read_hdf5_run_time, read_csv_file\n", "from Analyser.ImagingAnalyser import ImageAnalyser\n", "from Analyser.FitAnalyser import FitAnalyser\n", "from Analyser.FitAnalyser import ThomasFermi2dModel, DensityProfileBEC2dModel, Polylog22dModel\n", "from Analyser.FFTAnalyser import fft, ifft, fft_nutou\n", "from ToolFunction.ToolFunction import *\n", "\n", "from ToolFunction.HomeMadeXarrayFunction import errorbar, dataarray_plot_errorbar\n", "xr.plot.dataarray_plot.errorbar = errorbar\n", "xr.plot.accessor.DataArrayPlotAccessor.errorbar = dataarray_plot_errorbar\n", "\n", "imageAnalyser = ImageAnalyser()" ] }, { "attachments": {}, "cell_type": "markdown", "metadata": {}, "source": [ "# Import supporting package" ] }, { "cell_type": "code", "execution_count": 44, "metadata": {}, "outputs": [], "source": [ "import xarray as xr\n", "import numpy as np\n", "\n", "from uncertainties import ufloat\n", "from uncertainties import unumpy as unp\n", "from uncertainties import umath\n", "\n", "import matplotlib.pyplot as plt\n", "\n", "from DataContainer.ReadData import read_hdf5_file\n", "from Analyser.ImagingAnalyser import ImageAnalyser\n", "from Analyser.FitAnalyser import FitAnalyser\n", "from Analyser.FitAnalyser import ThomasFermi2dModel, DensityProfileBEC2dModel, Polylog22dModel\n", "from Analyser.FitAnalyser import NewFitModel\n", "from ToolFunction.ToolFunction import *\n", "\n", "from ToolFunction.HomeMadeXarrayFunction import errorbar, dataarray_plot_errorbar\n", "xr.plot.dataarray_plot.errorbar = errorbar\n", "xr.plot.accessor.DataArrayPlotAccessor.errorbar = dataarray_plot_errorbar\n", "\n", "imageAnalyser = ImageAnalyser()" ] }, { "attachments": {}, "cell_type": "markdown", "metadata": {}, "source": [ "## Start a client for parallel computing" ] }, { "cell_type": "code", "execution_count": 45, "metadata": {}, "outputs": [ { "data": { "text/html": [ "<div>\n", " <div style=\"width: 24px; height: 24px; background-color: #e1e1e1; border: 3px solid #9D9D9D; border-radius: 5px; position: absolute;\"> </div>\n", " <div style=\"margin-left: 48px;\">\n", " <h3 style=\"margin-bottom: 0px;\">Client</h3>\n", " <p style=\"color: #9D9D9D; margin-bottom: 0px;\">Client-f3431762-0b91-11ee-bc80-80e82ce2fa8e</p>\n", " <table style=\"width: 100%; text-align: left;\">\n", "\n", " <tr>\n", " \n", " <td style=\"text-align: left;\"><strong>Connection method:</strong> Cluster object</td>\n", " <td style=\"text-align: left;\"><strong>Cluster type:</strong> distributed.LocalCluster</td>\n", " \n", " </tr>\n", "\n", " \n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Dashboard: </strong> <a href=\"http://127.0.0.1:8787/status\" target=\"_blank\">http://127.0.0.1:8787/status</a>\n", " </td>\n", " <td style=\"text-align: left;\"></td>\n", " </tr>\n", " \n", "\n", " </table>\n", "\n", " \n", "\n", " \n", " <details>\n", " <summary style=\"margin-bottom: 20px;\"><h3 style=\"display: inline;\">Cluster Info</h3></summary>\n", " <div class=\"jp-RenderedHTMLCommon jp-RenderedHTML jp-mod-trusted jp-OutputArea-output\">\n", " <div style=\"width: 24px; height: 24px; background-color: #e1e1e1; border: 3px solid #9D9D9D; border-radius: 5px; position: absolute;\">\n", " </div>\n", " <div style=\"margin-left: 48px;\">\n", " <h3 style=\"margin-bottom: 0px; margin-top: 0px;\">LocalCluster</h3>\n", " <p style=\"color: #9D9D9D; margin-bottom: 0px;\">6e648e73</p>\n", " <table style=\"width: 100%; text-align: left;\">\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Dashboard:</strong> <a href=\"http://127.0.0.1:8787/status\" target=\"_blank\">http://127.0.0.1:8787/status</a>\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Workers:</strong> 6\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Total threads:</strong> 60\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Total memory:</strong> 55.88 GiB\n", " </td>\n", " </tr>\n", " \n", " <tr>\n", " <td style=\"text-align: left;\"><strong>Status:</strong> running</td>\n", " <td style=\"text-align: left;\"><strong>Using processes:</strong> True</td>\n", "</tr>\n", "\n", " \n", " </table>\n", "\n", " <details>\n", " <summary style=\"margin-bottom: 20px;\">\n", " <h3 style=\"display: inline;\">Scheduler Info</h3>\n", " </summary>\n", "\n", " <div style=\"\">\n", " <div>\n", " <div style=\"width: 24px; height: 24px; background-color: #FFF7E5; border: 3px solid #FF6132; border-radius: 5px; position: absolute;\"> </div>\n", " <div style=\"margin-left: 48px;\">\n", " <h3 style=\"margin-bottom: 0px;\">Scheduler</h3>\n", " <p style=\"color: #9D9D9D; margin-bottom: 0px;\">Scheduler-669a9b65-bae9-4798-96b7-f5d552eb72f9</p>\n", " <table style=\"width: 100%; text-align: left;\">\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Comm:</strong> tcp://127.0.0.1:51057\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Workers:</strong> 6\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Dashboard:</strong> <a href=\"http://127.0.0.1:8787/status\" target=\"_blank\">http://127.0.0.1:8787/status</a>\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Total threads:</strong> 60\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Started:</strong> Just now\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Total memory:</strong> 55.88 GiB\n", " </td>\n", " </tr>\n", " </table>\n", " </div>\n", " </div>\n", "\n", " <details style=\"margin-left: 48px;\">\n", " <summary style=\"margin-bottom: 20px;\">\n", " <h3 style=\"display: inline;\">Workers</h3>\n", " </summary>\n", "\n", " \n", " <div style=\"margin-bottom: 20px;\">\n", " <div style=\"width: 24px; height: 24px; background-color: #DBF5FF; border: 3px solid #4CC9FF; border-radius: 5px; position: absolute;\"> </div>\n", " <div style=\"margin-left: 48px;\">\n", " <details>\n", " <summary>\n", " <h4 style=\"margin-bottom: 0px; display: inline;\">Worker: 0</h4>\n", " </summary>\n", " <table style=\"width: 100%; text-align: left;\">\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Comm: </strong> tcp://127.0.0.1:51088\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Total threads: </strong> 10\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Dashboard: </strong> <a href=\"http://127.0.0.1:51093/status\" target=\"_blank\">http://127.0.0.1:51093/status</a>\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Memory: </strong> 9.31 GiB\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Nanny: </strong> tcp://127.0.0.1:51060\n", " </td>\n", " <td style=\"text-align: left;\"></td>\n", " </tr>\n", " <tr>\n", " <td colspan=\"2\" style=\"text-align: left;\">\n", " <strong>Local directory: </strong> C:\\Users\\data\\AppData\\Local\\Temp\\dask-worker-space\\worker-9s507mc2\n", " </td>\n", " </tr>\n", "\n", " \n", "\n", " \n", "\n", " </table>\n", " </details>\n", " </div>\n", " </div>\n", " \n", " <div style=\"margin-bottom: 20px;\">\n", " <div style=\"width: 24px; height: 24px; background-color: #DBF5FF; border: 3px solid #4CC9FF; border-radius: 5px; position: absolute;\"> </div>\n", " <div style=\"margin-left: 48px;\">\n", " <details>\n", " <summary>\n", " <h4 style=\"margin-bottom: 0px; display: inline;\">Worker: 1</h4>\n", " </summary>\n", " <table style=\"width: 100%; text-align: left;\">\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Comm: </strong> tcp://127.0.0.1:51084\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Total threads: </strong> 10\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Dashboard: </strong> <a href=\"http://127.0.0.1:51085/status\" target=\"_blank\">http://127.0.0.1:51085/status</a>\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Memory: </strong> 9.31 GiB\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Nanny: </strong> tcp://127.0.0.1:51061\n", " </td>\n", " <td style=\"text-align: left;\"></td>\n", " </tr>\n", " <tr>\n", " <td colspan=\"2\" style=\"text-align: left;\">\n", " <strong>Local directory: </strong> C:\\Users\\data\\AppData\\Local\\Temp\\dask-worker-space\\worker-y5skkt4c\n", " </td>\n", " </tr>\n", "\n", " \n", "\n", " \n", "\n", " </table>\n", " </details>\n", " </div>\n", " </div>\n", " \n", " <div style=\"margin-bottom: 20px;\">\n", " <div style=\"width: 24px; height: 24px; background-color: #DBF5FF; border: 3px solid #4CC9FF; border-radius: 5px; position: absolute;\"> </div>\n", " <div style=\"margin-left: 48px;\">\n", " <details>\n", " <summary>\n", " <h4 style=\"margin-bottom: 0px; display: inline;\">Worker: 2</h4>\n", " </summary>\n", " <table style=\"width: 100%; text-align: left;\">\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Comm: </strong> tcp://127.0.0.1:51098\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Total threads: </strong> 10\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Dashboard: </strong> <a href=\"http://127.0.0.1:51100/status\" target=\"_blank\">http://127.0.0.1:51100/status</a>\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Memory: </strong> 9.31 GiB\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Nanny: </strong> tcp://127.0.0.1:51062\n", " </td>\n", " <td style=\"text-align: left;\"></td>\n", " </tr>\n", " <tr>\n", " <td colspan=\"2\" style=\"text-align: left;\">\n", " <strong>Local directory: </strong> C:\\Users\\data\\AppData\\Local\\Temp\\dask-worker-space\\worker-gmddkxg0\n", " </td>\n", " </tr>\n", "\n", " \n", "\n", " \n", "\n", " </table>\n", " </details>\n", " </div>\n", " </div>\n", " \n", " <div style=\"margin-bottom: 20px;\">\n", " <div style=\"width: 24px; height: 24px; background-color: #DBF5FF; border: 3px solid #4CC9FF; border-radius: 5px; position: absolute;\"> </div>\n", " <div style=\"margin-left: 48px;\">\n", " <details>\n", " <summary>\n", " <h4 style=\"margin-bottom: 0px; display: inline;\">Worker: 3</h4>\n", " </summary>\n", " <table style=\"width: 100%; text-align: left;\">\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Comm: </strong> tcp://127.0.0.1:51095\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Total threads: </strong> 10\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Dashboard: </strong> <a href=\"http://127.0.0.1:51096/status\" target=\"_blank\">http://127.0.0.1:51096/status</a>\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Memory: </strong> 9.31 GiB\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Nanny: </strong> tcp://127.0.0.1:51063\n", " </td>\n", " <td style=\"text-align: left;\"></td>\n", " </tr>\n", " <tr>\n", " <td colspan=\"2\" style=\"text-align: left;\">\n", " <strong>Local directory: </strong> C:\\Users\\data\\AppData\\Local\\Temp\\dask-worker-space\\worker-oycines6\n", " </td>\n", " </tr>\n", "\n", " \n", "\n", " \n", "\n", " </table>\n", " </details>\n", " </div>\n", " </div>\n", " \n", " <div style=\"margin-bottom: 20px;\">\n", " <div style=\"width: 24px; height: 24px; background-color: #DBF5FF; border: 3px solid #4CC9FF; border-radius: 5px; position: absolute;\"> </div>\n", " <div style=\"margin-left: 48px;\">\n", " <details>\n", " <summary>\n", " <h4 style=\"margin-bottom: 0px; display: inline;\">Worker: 4</h4>\n", " </summary>\n", " <table style=\"width: 100%; text-align: left;\">\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Comm: </strong> tcp://127.0.0.1:51087\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Total threads: </strong> 10\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Dashboard: </strong> <a href=\"http://127.0.0.1:51091/status\" target=\"_blank\">http://127.0.0.1:51091/status</a>\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Memory: </strong> 9.31 GiB\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Nanny: </strong> tcp://127.0.0.1:51064\n", " </td>\n", " <td style=\"text-align: left;\"></td>\n", " </tr>\n", " <tr>\n", " <td colspan=\"2\" style=\"text-align: left;\">\n", " <strong>Local directory: </strong> C:\\Users\\data\\AppData\\Local\\Temp\\dask-worker-space\\worker-a8kpxp6o\n", " </td>\n", " </tr>\n", "\n", " \n", "\n", " \n", "\n", " </table>\n", " </details>\n", " </div>\n", " </div>\n", " \n", " <div style=\"margin-bottom: 20px;\">\n", " <div style=\"width: 24px; height: 24px; background-color: #DBF5FF; border: 3px solid #4CC9FF; border-radius: 5px; position: absolute;\"> </div>\n", " <div style=\"margin-left: 48px;\">\n", " <details>\n", " <summary>\n", " <h4 style=\"margin-bottom: 0px; display: inline;\">Worker: 5</h4>\n", " </summary>\n", " <table style=\"width: 100%; text-align: left;\">\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Comm: </strong> tcp://127.0.0.1:51099\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Total threads: </strong> 10\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Dashboard: </strong> <a href=\"http://127.0.0.1:51101/status\" target=\"_blank\">http://127.0.0.1:51101/status</a>\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Memory: </strong> 9.31 GiB\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Nanny: </strong> tcp://127.0.0.1:51065\n", " </td>\n", " <td style=\"text-align: left;\"></td>\n", " </tr>\n", " <tr>\n", " <td colspan=\"2\" style=\"text-align: left;\">\n", " <strong>Local directory: </strong> C:\\Users\\data\\AppData\\Local\\Temp\\dask-worker-space\\worker-thoxr07z\n", " </td>\n", " </tr>\n", "\n", " \n", "\n", " \n", "\n", " </table>\n", " </details>\n", " </div>\n", " </div>\n", " \n", "\n", " </details>\n", "</div>\n", "\n", " </details>\n", " </div>\n", "</div>\n", " </details>\n", " \n", "\n", " </div>\n", "</div>" ], "text/plain": [ "<Client: 'tcp://127.0.0.1:51057' processes=6 threads=60, memory=55.88 GiB>" ] }, "execution_count": 45, "metadata": {}, "output_type": "execute_result" } ], "source": [ "from dask.distributed import Client\n", "client = Client(n_workers=6, threads_per_worker=10, processes=True, memory_limit='10GB')\n", "client" ] }, { "attachments": {}, "cell_type": "markdown", "metadata": {}, "source": [ "## Set global path for experiment" ] }, { "cell_type": "code", "execution_count": 68, "metadata": {}, "outputs": [], "source": [ "# filepath = \"//DyLabNAS/Data/Evaporative_Cooling/2023/05/03/0043/*.h5\"\n", "# filepath = \"//DyLabNAS/Data/Evaporative_Cooling/2023/04/18/0003/2023-04-18_0003_Evaporative_Cooling_000.h5\"\n", "\n", "# filepath = \"//DyLabNAS/Data/Repetition_scan/2023/04/21/0002/*.h5\"\n", "\n", "# filepath = r\"./testData/0002/*.h5\"\n", "\n", "# filepath = r\"./testData/0002/2023-04-21_0002_Evaporative_Cooling_0.h5\"\n", "\n", "# filepath = r'd:/Jianshun Gao/Simulations/analyseScripts/testData/0002/2023-04-21_0002_Evaporative_Cooling_0.h5'\n", "\n", "# filepath = \"//DyLabNAS/Data/Evaporative_Cooling/2023/04/18/0003/*.h5\"\n", "\n", "filepath = \"//DyLabNAS/Data/Evaporative_Cooling/2023/05/04/0000/*.h5\"\n", "\n", "# filepath = './result_from_experiment/2023-04-24/0013/2023-04-24_0013_Evaporative_Cooling_13.h5'" ] }, { "cell_type": "code", "execution_count": 69, "metadata": {}, "outputs": [], "source": [ "groupList = [\n", " \"images/MOT_3D_Camera/in_situ_absorption\",\n", " \"images/ODT_1_Axis_Camera/in_situ_absorption\",\n", " \"images/ODT_2_Axis_Camera/in_situ_absorption\",\n", "]\n", "\n", "dskey = {\n", " \"images/MOT_3D_Camera/in_situ_absorption\": \"camera_1\",\n", " \"images/ODT_1_Axis_Camera/in_situ_absorption\": \"camera_2\",\n", " \"images/ODT_2_Axis_Camera/in_situ_absorption\": \"camera_3\",\n", "}\n" ] }, { "cell_type": "code", "execution_count": 70, "metadata": {}, "outputs": [], "source": [ "img_dir = '//DyLabNAS/Data/'\n", "SequenceName = \"Evaporative_Cooling\" + \"/\"\n", "folderPath = img_dir + SequenceName + '2023/05/23'# get_date()" ] }, { "attachments": {}, "cell_type": "markdown", "metadata": {}, "source": [ "# An example for one experimental run" ] }, { "attachments": {}, "cell_type": "markdown", "metadata": {}, "source": [ "## Load the data" ] }, { "cell_type": "code", "execution_count": 75, "metadata": {}, "outputs": [ { "name": "stderr", "output_type": "stream", "text": [ "f:\\Jianshun\\analyseScript\\DataContainer\\ReadData.py:178: FutureWarning: elementwise comparison failed; returning scalar instead, but in the future will perform elementwise comparison\n", " if not key in datesetOfGlobal.scanAxis\n" ] }, { "data": { "text/html": [ "<div><svg style=\"position: absolute; width: 0; height: 0; overflow: hidden\">\n", "<defs>\n", "<symbol id=\"icon-database\" viewBox=\"0 0 32 32\">\n", "<path d=\"M16 0c-8.837 0-16 2.239-16 5v4c0 2.761 7.163 5 16 5s16-2.239 16-5v-4c0-2.761-7.163-5-16-5z\"></path>\n", "<path d=\"M16 17c-8.837 0-16-2.239-16-5v6c0 2.761 7.163 5 16 5s16-2.239 16-5v-6c0 2.761-7.163 5-16 5z\"></path>\n", "<path d=\"M16 26c-8.837 0-16-2.239-16-5v6c0 2.761 7.163 5 16 5s16-2.239 16-5v-6c0 2.761-7.163 5-16 5z\"></path>\n", "</symbol>\n", "<symbol id=\"icon-file-text2\" viewBox=\"0 0 32 32\">\n", "<path d=\"M28.681 7.159c-0.694-0.947-1.662-2.053-2.724-3.116s-2.169-2.030-3.116-2.724c-1.612-1.182-2.393-1.319-2.841-1.319h-15.5c-1.378 0-2.5 1.121-2.5 2.5v27c0 1.378 1.122 2.5 2.5 2.5h23c1.378 0 2.5-1.122 2.5-2.5v-19.5c0-0.448-0.137-1.23-1.319-2.841zM24.543 5.457c0.959 0.959 1.712 1.825 2.268 2.543h-4.811v-4.811c0.718 0.556 1.584 1.309 2.543 2.268zM28 29.5c0 0.271-0.229 0.5-0.5 0.5h-23c-0.271 0-0.5-0.229-0.5-0.5v-27c0-0.271 0.229-0.5 0.5-0.5 0 0 15.499-0 15.5 0v7c0 0.552 0.448 1 1 1h7v19.5z\"></path>\n", "<path d=\"M23 26h-14c-0.552 0-1-0.448-1-1s0.448-1 1-1h14c0.552 0 1 0.448 1 1s-0.448 1-1 1z\"></path>\n", "<path d=\"M23 22h-14c-0.552 0-1-0.448-1-1s0.448-1 1-1h14c0.552 0 1 0.448 1 1s-0.448 1-1 1z\"></path>\n", "<path d=\"M23 18h-14c-0.552 0-1-0.448-1-1s0.448-1 1-1h14c0.552 0 1 0.448 1 1s-0.448 1-1 1z\"></path>\n", "</symbol>\n", "</defs>\n", "</svg>\n", "<style>/* CSS stylesheet for displaying xarray objects in jupyterlab.\n", " *\n", " */\n", "\n", ":root {\n", " --xr-font-color0: var(--jp-content-font-color0, rgba(0, 0, 0, 1));\n", " --xr-font-color2: var(--jp-content-font-color2, rgba(0, 0, 0, 0.54));\n", " --xr-font-color3: var(--jp-content-font-color3, rgba(0, 0, 0, 0.38));\n", " --xr-border-color: var(--jp-border-color2, #e0e0e0);\n", " --xr-disabled-color: var(--jp-layout-color3, #bdbdbd);\n", " --xr-background-color: var(--jp-layout-color0, white);\n", " --xr-background-color-row-even: var(--jp-layout-color1, white);\n", " --xr-background-color-row-odd: var(--jp-layout-color2, #eeeeee);\n", "}\n", "\n", "html[theme=dark],\n", "body[data-theme=dark],\n", "body.vscode-dark {\n", " --xr-font-color0: rgba(255, 255, 255, 1);\n", " --xr-font-color2: rgba(255, 255, 255, 0.54);\n", " --xr-font-color3: rgba(255, 255, 255, 0.38);\n", " --xr-border-color: #1F1F1F;\n", " --xr-disabled-color: #515151;\n", " --xr-background-color: #111111;\n", " --xr-background-color-row-even: #111111;\n", " --xr-background-color-row-odd: #313131;\n", "}\n", "\n", ".xr-wrap {\n", " display: block !important;\n", " min-width: 300px;\n", " max-width: 700px;\n", "}\n", "\n", ".xr-text-repr-fallback {\n", " /* fallback to plain text repr when CSS is not injected (untrusted notebook) */\n", " display: none;\n", "}\n", "\n", ".xr-header {\n", " padding-top: 6px;\n", " padding-bottom: 6px;\n", " margin-bottom: 4px;\n", " border-bottom: solid 1px var(--xr-border-color);\n", "}\n", "\n", ".xr-header > div,\n", ".xr-header > ul {\n", " display: inline;\n", " margin-top: 0;\n", " margin-bottom: 0;\n", "}\n", "\n", ".xr-obj-type,\n", ".xr-array-name {\n", " margin-left: 2px;\n", " margin-right: 10px;\n", "}\n", "\n", ".xr-obj-type {\n", " color: var(--xr-font-color2);\n", "}\n", "\n", ".xr-sections {\n", " padding-left: 0 !important;\n", " display: grid;\n", " grid-template-columns: 150px auto auto 1fr 20px 20px;\n", "}\n", "\n", ".xr-section-item {\n", " display: contents;\n", "}\n", "\n", ".xr-section-item input {\n", " display: none;\n", "}\n", "\n", ".xr-section-item input + label {\n", " color: var(--xr-disabled-color);\n", "}\n", "\n", ".xr-section-item input:enabled + label {\n", " cursor: pointer;\n", " color: var(--xr-font-color2);\n", "}\n", "\n", ".xr-section-item input:enabled + label:hover {\n", " color: var(--xr-font-color0);\n", "}\n", "\n", ".xr-section-summary {\n", " grid-column: 1;\n", " color: var(--xr-font-color2);\n", " font-weight: 500;\n", "}\n", "\n", ".xr-section-summary > span {\n", " display: inline-block;\n", " padding-left: 0.5em;\n", "}\n", "\n", ".xr-section-summary-in:disabled + label {\n", " color: var(--xr-font-color2);\n", "}\n", "\n", ".xr-section-summary-in + label:before {\n", " display: inline-block;\n", " content: 'â–º';\n", " font-size: 11px;\n", " width: 15px;\n", " text-align: center;\n", "}\n", "\n", ".xr-section-summary-in:disabled + label:before {\n", " color: var(--xr-disabled-color);\n", "}\n", "\n", ".xr-section-summary-in:checked + label:before {\n", " content: 'â–¼';\n", "}\n", "\n", ".xr-section-summary-in:checked + label > span {\n", " display: none;\n", "}\n", "\n", ".xr-section-summary,\n", ".xr-section-inline-details {\n", " padding-top: 4px;\n", " padding-bottom: 4px;\n", "}\n", "\n", ".xr-section-inline-details {\n", " grid-column: 2 / -1;\n", "}\n", "\n", ".xr-section-details {\n", " display: none;\n", " grid-column: 1 / -1;\n", " margin-bottom: 5px;\n", "}\n", "\n", ".xr-section-summary-in:checked ~ .xr-section-details {\n", " display: contents;\n", "}\n", "\n", ".xr-array-wrap {\n", " grid-column: 1 / -1;\n", " display: grid;\n", " grid-template-columns: 20px auto;\n", "}\n", "\n", ".xr-array-wrap > label {\n", " grid-column: 1;\n", " vertical-align: top;\n", "}\n", "\n", ".xr-preview {\n", " color: var(--xr-font-color3);\n", "}\n", "\n", ".xr-array-preview,\n", ".xr-array-data {\n", " padding: 0 5px !important;\n", " grid-column: 2;\n", "}\n", "\n", ".xr-array-data,\n", ".xr-array-in:checked ~ .xr-array-preview {\n", " display: none;\n", "}\n", "\n", ".xr-array-in:checked ~ .xr-array-data,\n", ".xr-array-preview {\n", " display: inline-block;\n", "}\n", "\n", ".xr-dim-list {\n", " display: inline-block !important;\n", " list-style: none;\n", " padding: 0 !important;\n", " margin: 0;\n", "}\n", "\n", ".xr-dim-list li {\n", " display: inline-block;\n", " padding: 0;\n", " margin: 0;\n", "}\n", "\n", ".xr-dim-list:before {\n", " content: '(';\n", "}\n", "\n", ".xr-dim-list:after {\n", " content: ')';\n", "}\n", "\n", ".xr-dim-list li:not(:last-child):after {\n", " content: ',';\n", " padding-right: 5px;\n", "}\n", "\n", ".xr-has-index {\n", " font-weight: bold;\n", "}\n", "\n", ".xr-var-list,\n", ".xr-var-item {\n", " display: contents;\n", "}\n", "\n", ".xr-var-item > div,\n", ".xr-var-item label,\n", ".xr-var-item > .xr-var-name span {\n", " background-color: var(--xr-background-color-row-even);\n", " margin-bottom: 0;\n", "}\n", "\n", ".xr-var-item > .xr-var-name:hover span {\n", " padding-right: 5px;\n", "}\n", "\n", ".xr-var-list > li:nth-child(odd) > div,\n", ".xr-var-list > li:nth-child(odd) > label,\n", ".xr-var-list > li:nth-child(odd) > .xr-var-name span {\n", " background-color: var(--xr-background-color-row-odd);\n", "}\n", "\n", ".xr-var-name {\n", " grid-column: 1;\n", "}\n", "\n", ".xr-var-dims {\n", " grid-column: 2;\n", "}\n", "\n", ".xr-var-dtype {\n", " grid-column: 3;\n", " text-align: right;\n", " color: var(--xr-font-color2);\n", "}\n", "\n", ".xr-var-preview {\n", " grid-column: 4;\n", "}\n", "\n", ".xr-index-preview {\n", " grid-column: 2 / 5;\n", " color: var(--xr-font-color2);\n", "}\n", "\n", ".xr-var-name,\n", ".xr-var-dims,\n", ".xr-var-dtype,\n", ".xr-preview,\n", ".xr-attrs dt {\n", " white-space: nowrap;\n", " overflow: hidden;\n", " text-overflow: ellipsis;\n", " padding-right: 10px;\n", "}\n", "\n", ".xr-var-name:hover,\n", ".xr-var-dims:hover,\n", ".xr-var-dtype:hover,\n", ".xr-attrs dt:hover {\n", " overflow: visible;\n", " width: auto;\n", " z-index: 1;\n", "}\n", "\n", ".xr-var-attrs,\n", ".xr-var-data,\n", ".xr-index-data {\n", " display: none;\n", " background-color: var(--xr-background-color) !important;\n", " padding-bottom: 5px !important;\n", "}\n", "\n", ".xr-var-attrs-in:checked ~ .xr-var-attrs,\n", ".xr-var-data-in:checked ~ .xr-var-data,\n", ".xr-index-data-in:checked ~ .xr-index-data {\n", " display: block;\n", "}\n", "\n", ".xr-var-data > table {\n", " float: right;\n", "}\n", "\n", ".xr-var-name span,\n", ".xr-var-data,\n", ".xr-index-name div,\n", ".xr-index-data,\n", ".xr-attrs {\n", " padding-left: 25px !important;\n", "}\n", "\n", ".xr-attrs,\n", ".xr-var-attrs,\n", ".xr-var-data,\n", ".xr-index-data {\n", " grid-column: 1 / -1;\n", "}\n", "\n", "dl.xr-attrs {\n", " padding: 0;\n", " margin: 0;\n", " display: grid;\n", " grid-template-columns: 125px auto;\n", "}\n", "\n", ".xr-attrs dt,\n", ".xr-attrs dd {\n", " padding: 0;\n", " margin: 0;\n", " float: left;\n", " padding-right: 10px;\n", " width: auto;\n", "}\n", "\n", ".xr-attrs dt {\n", " font-weight: normal;\n", " grid-column: 1;\n", "}\n", "\n", ".xr-attrs dt:hover span {\n", " display: inline-block;\n", " background: var(--xr-background-color);\n", " padding-right: 10px;\n", "}\n", "\n", ".xr-attrs dd {\n", " grid-column: 2;\n", " white-space: pre-wrap;\n", " word-break: break-all;\n", "}\n", "\n", ".xr-icon-database,\n", ".xr-icon-file-text2,\n", ".xr-no-icon {\n", " display: inline-block;\n", " vertical-align: middle;\n", " width: 1em;\n", " height: 1.5em !important;\n", " stroke-width: 0;\n", " stroke: currentColor;\n", " fill: currentColor;\n", "}\n", "</style><pre class='xr-text-repr-fallback'><xarray.Dataset>\n", "Dimensions: (y: 1200, x: 1920)\n", "Dimensions without coordinates: y, x\n", "Data variables:\n", " atoms (y, x) uint16 dask.array<chunksize=(1200, 1920), meta=np.ndarray>\n", " background (y, x) uint16 dask.array<chunksize=(1200, 1920), meta=np.ndarray>\n", " dark (y, x) uint16 dask.array<chunksize=(1200, 1920), meta=np.ndarray>\n", " shotNum <U2 '11'\n", " OD (y, x) float64 dask.array<chunksize=(1200, 1920), meta=np.ndarray>\n", "Attributes: (12/96)\n", " TOF_free: 0.02\n", " abs_img_freq: 110.858\n", " absorption_imaging_flag: True\n", " backup_data: True\n", " blink_off_time: nan\n", " blink_on_time: nan\n", " ... ...\n", " y_offset: 0\n", " y_offset_img: 0\n", " z_offset: 0.189\n", " z_offset_img: 0.189\n", " scanAxis: []\n", " scanAxisLength: []</pre><div class='xr-wrap' style='display:none'><div class='xr-header'><div class='xr-obj-type'>xarray.Dataset</div></div><ul class='xr-sections'><li class='xr-section-item'><input id='section-3f6a2152-ef01-4819-91f5-a47fe1a85483' class='xr-section-summary-in' type='checkbox' disabled ><label for='section-3f6a2152-ef01-4819-91f5-a47fe1a85483' class='xr-section-summary' title='Expand/collapse section'>Dimensions:</label><div class='xr-section-inline-details'><ul class='xr-dim-list'><li><span>y</span>: 1200</li><li><span>x</span>: 1920</li></ul></div><div class='xr-section-details'></div></li><li class='xr-section-item'><input id='section-71114a49-5445-43f4-a531-d6bdc6ba83e9' class='xr-section-summary-in' type='checkbox' disabled ><label for='section-71114a49-5445-43f4-a531-d6bdc6ba83e9' class='xr-section-summary' title='Expand/collapse section'>Coordinates: <span>(0)</span></label><div class='xr-section-inline-details'></div><div class='xr-section-details'><ul class='xr-var-list'></ul></div></li><li class='xr-section-item'><input id='section-87511f20-517f-4220-90f4-f268366d6d73' class='xr-section-summary-in' type='checkbox' checked><label for='section-87511f20-517f-4220-90f4-f268366d6d73' class='xr-section-summary' >Data variables: <span>(5)</span></label><div class='xr-section-inline-details'></div><div class='xr-section-details'><ul class='xr-var-list'><li class='xr-var-item'><div class='xr-var-name'><span>atoms</span></div><div class='xr-var-dims'>(y, x)</div><div class='xr-var-dtype'>uint16</div><div class='xr-var-preview xr-preview'>dask.array<chunksize=(1200, 1920), meta=np.ndarray></div><input id='attrs-81526946-1703-4ca5-8e13-3fc13c175996' class='xr-var-attrs-in' type='checkbox' ><label for='attrs-81526946-1703-4ca5-8e13-3fc13c175996' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-77f2b9fc-57a9-4fc9-bcb9-7283191c0038' class='xr-var-data-in' type='checkbox'><label for='data-77f2b9fc-57a9-4fc9-bcb9-7283191c0038' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'><dt><span>IMAGE_SUBCLASS :</span></dt><dd>IMAGE_GRAYSCALE</dd><dt><span>IMAGE_VERSION :</span></dt><dd>1.2</dd><dt><span>IMAGE_WHITE_IS_ZERO :</span></dt><dd>0</dd></dl></div><div class='xr-var-data'><table>\n", " <tr>\n", " <td>\n", " <table style=\"border-collapse: collapse;\">\n", " <thead>\n", " <tr>\n", " <td> </td>\n", " <th> Array </th>\n", " <th> Chunk </th>\n", " </tr>\n", " </thead>\n", " <tbody>\n", " \n", " <tr>\n", " <th> Bytes </th>\n", " <td> 4.39 MiB </td>\n", " <td> 4.39 MiB </td>\n", " </tr>\n", " \n", " <tr>\n", " <th> Shape </th>\n", " <td> (1200, 1920) </td>\n", " <td> (1200, 1920) </td>\n", " </tr>\n", " <tr>\n", " <th> Dask graph </th>\n", " <td colspan=\"2\"> 1 chunks in 2 graph layers </td>\n", " </tr>\n", " <tr>\n", " <th> Data type </th>\n", " <td colspan=\"2\"> uint16 numpy.ndarray </td>\n", " </tr>\n", " </tbody>\n", " </table>\n", " </td>\n", " <td>\n", " <svg width=\"170\" height=\"125\" style=\"stroke:rgb(0,0,0);stroke-width:1\" >\n", "\n", " <!-- Horizontal lines -->\n", " <line x1=\"0\" y1=\"0\" x2=\"120\" y2=\"0\" style=\"stroke-width:2\" />\n", " <line x1=\"0\" y1=\"75\" x2=\"120\" y2=\"75\" style=\"stroke-width:2\" />\n", "\n", " <!-- Vertical lines -->\n", " <line x1=\"0\" y1=\"0\" x2=\"0\" y2=\"75\" style=\"stroke-width:2\" />\n", " <line x1=\"120\" y1=\"0\" x2=\"120\" y2=\"75\" style=\"stroke-width:2\" />\n", "\n", " <!-- Colored Rectangle -->\n", " <polygon points=\"0.0,0.0 120.0,0.0 120.0,75.0 0.0,75.0\" style=\"fill:#ECB172A0;stroke-width:0\"/>\n", "\n", " <!-- Text -->\n", " <text x=\"60.000000\" y=\"95.000000\" font-size=\"1.0rem\" font-weight=\"100\" text-anchor=\"middle\" >1920</text>\n", " <text x=\"140.000000\" y=\"37.500000\" font-size=\"1.0rem\" font-weight=\"100\" text-anchor=\"middle\" transform=\"rotate(-90,140.000000,37.500000)\">1200</text>\n", "</svg>\n", " </td>\n", " </tr>\n", "</table></div></li><li class='xr-var-item'><div class='xr-var-name'><span>background</span></div><div class='xr-var-dims'>(y, x)</div><div class='xr-var-dtype'>uint16</div><div class='xr-var-preview xr-preview'>dask.array<chunksize=(1200, 1920), meta=np.ndarray></div><input id='attrs-cf0352d3-81af-4aa1-8a4b-d510a6d9ed9f' class='xr-var-attrs-in' type='checkbox' ><label for='attrs-cf0352d3-81af-4aa1-8a4b-d510a6d9ed9f' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-7cebfcb4-0749-41e1-a9df-8a3ca0e9cd57' class='xr-var-data-in' type='checkbox'><label for='data-7cebfcb4-0749-41e1-a9df-8a3ca0e9cd57' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'><dt><span>IMAGE_SUBCLASS :</span></dt><dd>IMAGE_GRAYSCALE</dd><dt><span>IMAGE_VERSION :</span></dt><dd>1.2</dd><dt><span>IMAGE_WHITE_IS_ZERO :</span></dt><dd>0</dd></dl></div><div class='xr-var-data'><table>\n", " <tr>\n", " <td>\n", " <table style=\"border-collapse: collapse;\">\n", " <thead>\n", " <tr>\n", " <td> </td>\n", " <th> Array </th>\n", " <th> Chunk </th>\n", " </tr>\n", " </thead>\n", " <tbody>\n", " \n", " <tr>\n", " <th> Bytes </th>\n", " <td> 4.39 MiB </td>\n", " <td> 4.39 MiB </td>\n", " </tr>\n", " \n", " <tr>\n", " <th> Shape </th>\n", " <td> (1200, 1920) </td>\n", " <td> (1200, 1920) </td>\n", " </tr>\n", " <tr>\n", " <th> Dask graph </th>\n", " <td colspan=\"2\"> 1 chunks in 2 graph layers </td>\n", " </tr>\n", " <tr>\n", " <th> Data type </th>\n", " <td colspan=\"2\"> uint16 numpy.ndarray </td>\n", " </tr>\n", " </tbody>\n", " </table>\n", " </td>\n", " <td>\n", " <svg width=\"170\" height=\"125\" style=\"stroke:rgb(0,0,0);stroke-width:1\" >\n", "\n", " <!-- Horizontal lines -->\n", " <line x1=\"0\" y1=\"0\" x2=\"120\" y2=\"0\" style=\"stroke-width:2\" />\n", " <line x1=\"0\" y1=\"75\" x2=\"120\" y2=\"75\" style=\"stroke-width:2\" />\n", "\n", " <!-- Vertical lines -->\n", " <line x1=\"0\" y1=\"0\" x2=\"0\" y2=\"75\" style=\"stroke-width:2\" />\n", " <line x1=\"120\" y1=\"0\" x2=\"120\" y2=\"75\" style=\"stroke-width:2\" />\n", "\n", " <!-- Colored Rectangle -->\n", " <polygon points=\"0.0,0.0 120.0,0.0 120.0,75.0 0.0,75.0\" style=\"fill:#ECB172A0;stroke-width:0\"/>\n", "\n", " <!-- Text -->\n", " <text x=\"60.000000\" y=\"95.000000\" font-size=\"1.0rem\" font-weight=\"100\" text-anchor=\"middle\" >1920</text>\n", " <text x=\"140.000000\" y=\"37.500000\" font-size=\"1.0rem\" font-weight=\"100\" text-anchor=\"middle\" transform=\"rotate(-90,140.000000,37.500000)\">1200</text>\n", "</svg>\n", " </td>\n", " </tr>\n", "</table></div></li><li class='xr-var-item'><div class='xr-var-name'><span>dark</span></div><div class='xr-var-dims'>(y, x)</div><div class='xr-var-dtype'>uint16</div><div class='xr-var-preview xr-preview'>dask.array<chunksize=(1200, 1920), meta=np.ndarray></div><input id='attrs-ea5ca203-117f-4d7f-a382-5ea49ef757fe' class='xr-var-attrs-in' type='checkbox' ><label for='attrs-ea5ca203-117f-4d7f-a382-5ea49ef757fe' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-1c46121e-396f-4c19-b720-5bfd85d3db08' class='xr-var-data-in' type='checkbox'><label for='data-1c46121e-396f-4c19-b720-5bfd85d3db08' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'><dt><span>IMAGE_SUBCLASS :</span></dt><dd>IMAGE_GRAYSCALE</dd><dt><span>IMAGE_VERSION :</span></dt><dd>1.2</dd><dt><span>IMAGE_WHITE_IS_ZERO :</span></dt><dd>0</dd></dl></div><div class='xr-var-data'><table>\n", " <tr>\n", " <td>\n", " <table style=\"border-collapse: collapse;\">\n", " <thead>\n", " <tr>\n", " <td> </td>\n", " <th> Array </th>\n", " <th> Chunk </th>\n", " </tr>\n", " </thead>\n", " <tbody>\n", " \n", " <tr>\n", " <th> Bytes </th>\n", " <td> 4.39 MiB </td>\n", " <td> 4.39 MiB </td>\n", " </tr>\n", " \n", " <tr>\n", " <th> Shape </th>\n", " <td> (1200, 1920) </td>\n", " <td> (1200, 1920) </td>\n", " </tr>\n", " <tr>\n", " <th> Dask graph </th>\n", " <td colspan=\"2\"> 1 chunks in 2 graph layers </td>\n", " </tr>\n", " <tr>\n", " <th> Data type </th>\n", " <td colspan=\"2\"> uint16 numpy.ndarray </td>\n", " </tr>\n", " </tbody>\n", " </table>\n", " </td>\n", " <td>\n", " <svg width=\"170\" height=\"125\" style=\"stroke:rgb(0,0,0);stroke-width:1\" >\n", "\n", " <!-- Horizontal lines -->\n", " <line x1=\"0\" y1=\"0\" x2=\"120\" y2=\"0\" style=\"stroke-width:2\" />\n", " <line x1=\"0\" y1=\"75\" x2=\"120\" y2=\"75\" style=\"stroke-width:2\" />\n", "\n", " <!-- Vertical lines -->\n", " <line x1=\"0\" y1=\"0\" x2=\"0\" y2=\"75\" style=\"stroke-width:2\" />\n", " <line x1=\"120\" y1=\"0\" x2=\"120\" y2=\"75\" style=\"stroke-width:2\" />\n", "\n", " <!-- Colored Rectangle -->\n", " <polygon points=\"0.0,0.0 120.0,0.0 120.0,75.0 0.0,75.0\" style=\"fill:#ECB172A0;stroke-width:0\"/>\n", "\n", " <!-- Text -->\n", " <text x=\"60.000000\" y=\"95.000000\" font-size=\"1.0rem\" font-weight=\"100\" text-anchor=\"middle\" >1920</text>\n", " <text x=\"140.000000\" y=\"37.500000\" font-size=\"1.0rem\" font-weight=\"100\" text-anchor=\"middle\" transform=\"rotate(-90,140.000000,37.500000)\">1200</text>\n", "</svg>\n", " </td>\n", " </tr>\n", "</table></div></li><li class='xr-var-item'><div class='xr-var-name'><span>shotNum</span></div><div class='xr-var-dims'>()</div><div class='xr-var-dtype'><U2</div><div class='xr-var-preview xr-preview'>'11'</div><input id='attrs-931dff3b-9054-49ae-a9a3-af23a3abe9d2' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-931dff3b-9054-49ae-a9a3-af23a3abe9d2' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-16b7a2e7-307c-45ef-8fd7-4961a352c6cd' class='xr-var-data-in' type='checkbox'><label for='data-16b7a2e7-307c-45ef-8fd7-4961a352c6cd' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array('11', dtype='<U2')</pre></div></li><li class='xr-var-item'><div class='xr-var-name'><span>OD</span></div><div class='xr-var-dims'>(y, x)</div><div class='xr-var-dtype'>float64</div><div class='xr-var-preview xr-preview'>dask.array<chunksize=(1200, 1920), meta=np.ndarray></div><input id='attrs-c45c4a1c-0006-43c9-be36-4e3d40196cf6' class='xr-var-attrs-in' type='checkbox' ><label for='attrs-c45c4a1c-0006-43c9-be36-4e3d40196cf6' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-81f04315-1fc1-4433-9b5f-8dbcef03d3d4' class='xr-var-data-in' type='checkbox'><label for='data-81f04315-1fc1-4433-9b5f-8dbcef03d3d4' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'><dt><span>IMAGE_SUBCLASS :</span></dt><dd>IMAGE_GRAYSCALE</dd><dt><span>IMAGE_VERSION :</span></dt><dd>1.2</dd><dt><span>IMAGE_WHITE_IS_ZERO :</span></dt><dd>0</dd></dl></div><div class='xr-var-data'><table>\n", " <tr>\n", " <td>\n", " <table style=\"border-collapse: collapse;\">\n", " <thead>\n", " <tr>\n", " <td> </td>\n", " <th> Array </th>\n", " <th> Chunk </th>\n", " </tr>\n", " </thead>\n", " <tbody>\n", " \n", " <tr>\n", " <th> Bytes </th>\n", " <td> 17.58 MiB </td>\n", " <td> 17.58 MiB </td>\n", " </tr>\n", " \n", " <tr>\n", " <th> Shape </th>\n", " <td> (1200, 1920) </td>\n", " <td> (1200, 1920) </td>\n", " </tr>\n", " <tr>\n", " <th> Dask graph </th>\n", " <td colspan=\"2\"> 1 chunks in 16 graph layers </td>\n", " </tr>\n", " <tr>\n", " <th> Data type </th>\n", " <td colspan=\"2\"> float64 numpy.ndarray </td>\n", " </tr>\n", " </tbody>\n", " </table>\n", " </td>\n", " <td>\n", " <svg width=\"170\" height=\"125\" style=\"stroke:rgb(0,0,0);stroke-width:1\" >\n", "\n", " <!-- Horizontal lines -->\n", " <line x1=\"0\" y1=\"0\" x2=\"120\" y2=\"0\" style=\"stroke-width:2\" />\n", " <line x1=\"0\" y1=\"75\" x2=\"120\" y2=\"75\" style=\"stroke-width:2\" />\n", "\n", " <!-- Vertical lines -->\n", " <line x1=\"0\" y1=\"0\" x2=\"0\" y2=\"75\" style=\"stroke-width:2\" />\n", " <line x1=\"120\" y1=\"0\" x2=\"120\" y2=\"75\" style=\"stroke-width:2\" />\n", "\n", " <!-- Colored Rectangle -->\n", " <polygon points=\"0.0,0.0 120.0,0.0 120.0,75.0 0.0,75.0\" style=\"fill:#ECB172A0;stroke-width:0\"/>\n", "\n", " <!-- Text -->\n", " <text x=\"60.000000\" y=\"95.000000\" font-size=\"1.0rem\" font-weight=\"100\" text-anchor=\"middle\" >1920</text>\n", " <text x=\"140.000000\" y=\"37.500000\" font-size=\"1.0rem\" font-weight=\"100\" text-anchor=\"middle\" transform=\"rotate(-90,140.000000,37.500000)\">1200</text>\n", "</svg>\n", " </td>\n", " </tr>\n", "</table></div></li></ul></div></li><li class='xr-section-item'><input id='section-c33811d7-8cf5-4e35-8e55-5bc202a3ba60' class='xr-section-summary-in' type='checkbox' disabled ><label for='section-c33811d7-8cf5-4e35-8e55-5bc202a3ba60' class='xr-section-summary' title='Expand/collapse section'>Indexes: <span>(0)</span></label><div class='xr-section-inline-details'></div><div class='xr-section-details'><ul class='xr-var-list'></ul></div></li><li class='xr-section-item'><input id='section-47715f26-a2f1-48c1-9cd1-23dbb2fa5488' class='xr-section-summary-in' type='checkbox' ><label for='section-47715f26-a2f1-48c1-9cd1-23dbb2fa5488' class='xr-section-summary' >Attributes: <span>(96)</span></label><div class='xr-section-inline-details'></div><div class='xr-section-details'><dl class='xr-attrs'><dt><span>TOF_free :</span></dt><dd>0.02</dd><dt><span>abs_img_freq :</span></dt><dd>110.858</dd><dt><span>absorption_imaging_flag :</span></dt><dd>True</dd><dt><span>backup_data :</span></dt><dd>True</dd><dt><span>blink_off_time :</span></dt><dd>nan</dd><dt><span>blink_on_time :</span></dt><dd>nan</dd><dt><span>c_duration :</span></dt><dd>0.2</dd><dt><span>cmot_final_current :</span></dt><dd>0.65</dd><dt><span>cmot_hold :</span></dt><dd>0.06</dd><dt><span>cmot_initial_current :</span></dt><dd>0.18</dd><dt><span>compX_current :</span></dt><dd>0.005</dd><dt><span>compX_current_sg :</span></dt><dd>0</dd><dt><span>compX_final_current :</span></dt><dd>0.005</dd><dt><span>compX_initial_current :</span></dt><dd>0.005</dd><dt><span>compY_current :</span></dt><dd>0</dd><dt><span>compY_current_sg :</span></dt><dd>0</dd><dt><span>compY_final_current :</span></dt><dd>0.0</dd><dt><span>compY_initial_current :</span></dt><dd>0</dd><dt><span>compZ_current :</span></dt><dd>0</dd><dt><span>compZ_current_sg :</span></dt><dd>0.189</dd><dt><span>compZ_final_current :</span></dt><dd>0.2812</dd><dt><span>compZ_initial_current :</span></dt><dd>0</dd><dt><span>default_camera :</span></dt><dd>0</dd><dt><span>evap_1_arm_1_final_pow :</span></dt><dd>0.35</dd><dt><span>evap_1_arm_1_mod_depth_final :</span></dt><dd>0</dd><dt><span>evap_1_arm_1_mod_depth_initial :</span></dt><dd>1.0</dd><dt><span>evap_1_arm_1_mod_ramp_duration :</span></dt><dd>1.15</dd><dt><span>evap_1_arm_1_pow_ramp_duration :</span></dt><dd>1.65</dd><dt><span>evap_1_arm_1_start_pow :</span></dt><dd>7</dd><dt><span>evap_1_arm_2_final_pow :</span></dt><dd>5</dd><dt><span>evap_1_arm_2_ramp_duration :</span></dt><dd>0.5</dd><dt><span>evap_1_arm_2_start_pow :</span></dt><dd>0</dd><dt><span>evap_1_mod_ramp_trunc_value :</span></dt><dd>1</dd><dt><span>evap_1_pow_ramp_trunc_value :</span></dt><dd>1.0</dd><dt><span>evap_1_rate_constant_1 :</span></dt><dd>0.525</dd><dt><span>evap_1_rate_constant_2 :</span></dt><dd>0.51</dd><dt><span>evap_2_arm_1_final_pow :</span></dt><dd>0.037</dd><dt><span>evap_2_arm_1_start_pow :</span></dt><dd>0.35</dd><dt><span>evap_2_arm_2_final_pow :</span></dt><dd>0.09</dd><dt><span>evap_2_arm_2_start_pow :</span></dt><dd>5</dd><dt><span>evap_2_ramp_duration :</span></dt><dd>1.0</dd><dt><span>evap_2_ramp_trunc_value :</span></dt><dd>1</dd><dt><span>evap_2_rate_constant_1 :</span></dt><dd>0.37</dd><dt><span>evap_2_rate_constant_2 :</span></dt><dd>0.71</dd><dt><span>evap_3_arm_1_final_pow :</span></dt><dd>0.1038</dd><dt><span>evap_3_arm_1_mod_depth_final :</span></dt><dd>0.43</dd><dt><span>evap_3_arm_1_mod_depth_initial :</span></dt><dd>0</dd><dt><span>evap_3_arm_1_start_pow :</span></dt><dd>0.037</dd><dt><span>evap_3_ramp_duration :</span></dt><dd>0.1</dd><dt><span>evap_3_ramp_trunc_value :</span></dt><dd>1</dd><dt><span>evap_3_rate_constant_1 :</span></dt><dd>-0.879</dd><dt><span>evap_3_rate_constant_2 :</span></dt><dd>-0.297</dd><dt><span>final_amp :</span></dt><dd>8e-05</dd><dt><span>final_freq :</span></dt><dd>104.0</dd><dt><span>gradCoil_current :</span></dt><dd>0.18</dd><dt><span>gradCoil_current_sg :</span></dt><dd>0</dd><dt><span>imaging_method :</span></dt><dd>in_situ_absorption</dd><dt><span>imaging_pulse_duration :</span></dt><dd>2.5e-05</dd><dt><span>imaging_wavel ], "text/plain": [ "<xarray.Dataset>\n", "Dimensions: (y: 1200, x: 1920)\n", "Dimensions without coordinates: y, x\n", "Data variables:\n", " atoms (y, x) uint16 dask.array<chunksize=(1200, 1920), meta=np.ndarray>\n", " background (y, x) uint16 dask.array<chunksize=(1200, 1920), meta=np.ndarray>\n", " dark (y, x) uint16 dask.array<chunksize=(1200, 1920), meta=np.ndarray>\n", " shotNum <U2 '11'\n", " OD (y, x) float64 dask.array<chunksize=(1200, 1920), meta=np.ndarray>\n", "Attributes: (12/96)\n", " TOF_free: 0.02\n", " abs_img_freq: 110.858\n", " absorption_imaging_flag: True\n", " backup_data: True\n", " blink_off_time: nan\n", " blink_on_time: nan\n", " ... ...\n", " y_offset: 0\n", " y_offset_img: 0\n", " z_offset: 0.189\n", " z_offset_img: 0.189\n", " scanAxis: []\n", " scanAxisLength: []" ] }, "execution_count": 75, "metadata": {}, "output_type": "execute_result" } ], "source": [ "shotNum = \"0069\"\n", "filePath = folderPath + \"/\" + shotNum + \"/*.h5\"\n", "# filePath = \"//DyLabNAS/Data/Evaporative_Cooling/2023/05/12/0065/*.h5\"\n", "filePath = './result_from_experiment/2023-04-24/0013/2023-04-24_0013_Evaporative_Cooling_11.h5'\n", "\n", "dataSetDict = {\n", " dskey[groupList[i]]: read_hdf5_file(filePath, groupList[i])\n", " for i in [0] # range(len(groupList))\n", "}\n", "\n", "dataSet = dataSetDict[\"camera_1\"]\n", "dataSet = swap_xy(dataSet)\n", "\n", "scanAxis = get_scanAxis(dataSet)\n", "\n", "dataSet = auto_rechunk(dataSet)\n", "\n", "dataSet = imageAnalyser.get_absorption_images(dataSet)\n", "\n", "dataSet" ] }, { "attachments": {}, "cell_type": "markdown", "metadata": {}, "source": [ "## Calculate an plot OD images" ] }, { "cell_type": "code", "execution_count": 76, "metadata": {}, "outputs": [ { "data": { "image/png": "iVBORw0KGgoAAAANSUhEUgAAAnIAAAHECAYAAAC9VcdgAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjUuMSwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy/YYfK9AAAACXBIWXMAAA9hAAAPYQGoP6dpAAEAAElEQVR4nOz9fZhWV33vj78muTFOzBAZcRCBk8EOkdEMp0zzBe2UMrnOqNDfmV4HrETJsZhCNbSHU9FINfT0ds6RpGJqzrdpJbVBjTGxiZa04ldiOl4ZGmkKTUhPRh00o5AQJFAyGIZkotzJ/P5Y632vz1p730MSnj37c11czL733muv5/Ve78/DqhsdHR2lkEIKKaSQQgoppJBzTs470xkopJBCCimkkEIKKeSVSQHkCimkkEIKKaSQQs5RKYBcIYUUUkghhRRSyDkqBZArpJBCCimkkEIKOUelAHKFFFJIIYUUUkgh56gUQK6QQgoppJBCCinkHJUCyBVSSCGFFFJIIYWco1I60xn4v11efPFFfvrTn9LQ0EBdXd2Zzk4hhRRSSCFnsYyOjjI8PMwb3/hGzjvv1HAxf//3f8/jjz9+UtJ673vfy6RJk05KWoXkSwHkzrD89Kc/Zdq0aWc6G4UUUkghhZxDsnfvXqZOnXrS0x0eHmbRokVcyomr7J7CrXGf/vSnT0LOCqklBZA7w9LQ0OD/+hbwGuAfgQrwRv/7T/3/VwAPmDcrwOv9303A9829pcCdwCf89Q/NP4CPATeYbzQB3wM+5K//2qe/0F9vSXLdDTQQus+X/f9vBX4jpHH+J+CFG5J3bZpvAw76658Al/r/fR67gN7PJGX+sP/7fzO2XAocBv49+b3k09EzPwEm+OvD5h7AYuDNuLqyzy/z1w8AP8r5rv/ttZ+An/2dKdMEeNMK+InKpG+pnW7Atem/wyz/26MAd5k0Xg+sMHlaCDyUlPNtwL/klFfyJpOe0tT7S/3//+j/V5281V/bfvZW88zB5Dt531V//fecPOgdgCW4/mtlObCRfEm/9SbgCf+3fn89MA94tb/+Ea7e0vyl/aWWvJW4LtJ31Q8+7K//98tI/x3+//uBywnj9AH//qX+upHQzva7h/3felZ9VPWketYc8u/mvuYGcP3/eya9CYQxZdvSLiNN8Opl8Pxt/vqnhLoH+Ib/rvrZnUkeVDbl+Qrg34Bhf93g/6l9la9r/N83JOWTLAY2+b/1PZXhMl/OFSaNtK3eSpg/NR5+21//GBgJj/7OZXAI6LPzxnhCfyvh2nSYMPf8BPhV3KQHMA74jCnDmwh9bgT4I7N2nFzRYU9LCKPllco3TXqFnDopgNwZlqBOfQ1wEW7oVIB6//urzX07rOwzFyb3Gvz1eH99UfxM3XgYfXXO+3peebgoyYPkImIgp/sXxmnUja/xrt55jX9H17Yc491climzzeNYciHwfM5zdtHX91QPI8ST/0X+e69Onm9IrtPvmnqOnqmH82x6+pYtU737/3z9ln6nPsnTReGdqti+kgeo0nzb9xvMbxDqxLaTTQdCPR8PyNk+nVd36k8NOffyfqv1LZu2fq8n1FX6jM3fS126xqpDez+nbY8rrzHvaF6w37jQPJfX/54379t85gG5+uR+2rdsenZM2ba0y0i97/f2vupe1xVCP0vrJe1nr/H3j5k6SOtRYyLNTzqW7RiyZdA3xmqrtE/Zue4i4Pzw6LjxPgu2TGk71JMdVzbNccRj6kJCezgpTHEKkRRA7qyTRuADwHrzWwnoNddNwCzggLusXwwj/ea+dsM9/v82YADHpAGj+r3R//8o0Ars89cdQItJpwnHuPj32ez/txNlB3AE2nya/UClx9xvAQaBbebbl/h8gdtR34LbBwKsgy120ZkF7AQ+Syy63+nLIYZvCJjk/9f3Utao1d9XPXyAuN6PmfetaPe9C0plU86yy7fkcE/82qKVcM8GsgDHPncQaIJH7vDXg8AcYGb4JnsI5d5MdhiXgFX+77twbIRsVHb6d1pM+gfDq9OaYW+SbyDUke8L08qwVwzHAbIAoRu6Z/ks9gDzkzIOjvGNAWJwNgXXN1JRGW/2/9s6AtfvNS4OAn3mG7uSd5qJ62IBTJ4L+/WtHri8DA+p//a5/0pl938lrbNOl95kDxb2U0OazN9DOLZXzONMwngBWA3cZPKepnEQt9Db+7uS+wBz/f/biNqeRhxrpXa8G1eHemYlsMHfX+l/6yFmu/bBSI9Jox3X5/r8dQU3TnqS+5LXAduJ55oOc38brp7mmPSO+HyBmz+OEM+XKovkYPJ/+qzuTQllopXQl/B/a+5K7n11ANeX1Oe3JmlXgMeBcsCzw+tw9aC6uAYH6jT/DPq/hwjA+tRKCb+XPgEpvClPjxRA7qyRLYQdWwpWJJpYGv0/PwGNtBE3pRZAgaK7CbtA/N+LCYtEBbcY24lqq0lTE56etxM3hAWmBP0GyNBEvHDOAXaY+1Z91uu/1+qvNTmqLJrgBEj24Ra6K/11H/GidJCw8Oh6GXAHWTCniXYXcB1BjfOoL7NVA0Gox5luAa/zi/loDzF46DLlAu7ZnuSxKblehQNes4AnzTMjxIAyZaDUH8DVz1bCIjNEDILaTBo5eTjky1X93pD/lu0rwF5COevLfvG2edoBm+8x1/XAvcSiBfoAYaHC52cmWcCSigCcgIH6ht47QFBV9fr7i/212kpyL7AIuCo8v38TgQkCHrrLpN0ELIRKCvQFDAeARtgvwLKMsDHC9ZnRdUR131aGfgsIBcL0zM1kxY77BTn3Me+rTg04nFeGBzSWF+P6n0BUIw5UCgjeZdKy+RzC9VkI41R94ZjPo+3n9t1Hifuz8qbN4ipcueeYd/SbZKb52wI2ifqZ0p4PbHX1DdA/AnzW7G1afJ52mectwG3EAXXNVfqG8qz3UgCneuzG1etOGN4cbreWYUBmLLcQNreSIZ+H07Nsn4wvvRLO8LnnnmPr1q08/PDD7Ny5k4cffpgnnnCq9HK5zCc/+ckTzBUcOHCA9evX881vfpMnnniC+vp63vrWt7Js2TKWL19+zrGdBZA762Q+bsKRzccthEmu2/x2gGCrNUK8K2zCLVp+UltUhns2E7NTdsK7DrjeXGvx0K77ZtwioYX4buKFVpPPXNxu2qaphRT/fU26B3E7Zws6h3FsE74sXVS7aMNcGN5AYA2bcIv/Xf46XfSXwkTgkF1ABEpW+2uxXraubT1ItGP3C0B1n9rh0hy14NXuYYdw4FuA4V6YXDaLu80bwNPme4v83/fg2ssCho048KJnLUBs9/8LkNzk86lFrJ942Cd5GNmJA8dakLXQKL07cO3sGSqAhzb5Mh4wae4j3nhYECcbTss2lam2x+SrYP8dBHDaTmh3iQU4qv904TxIzLYcIABk9U+BnxZcXVu22dYbZFmW24gYtboyjOp5fVf9XX1/hrscTZnZOR5QQGiflAUS66l5YCpuo2IZJstuqY6U54GkTCV4wICq0mKoWECSMlcHCaBL5W4ly2jZ/qVNhu1nU+LH2UeYy24j3gypPNoAziQLaGsBfoG/EnHf2ApcB/2P+ut7gCYY1KavEddOr/PXnbg683PZ9Lmw2475tJ3STUgTTFgJh/WOAL1l0yswsI64T9xCtZ6nr4TdPbix92yN8v5yyI4dO/it3/qtU5b+ww8/zLve9S6eftrNtxdddBHDw8N897vf5bvf/S5f+9rX+MY3vsEFF1xwyvJwsqUAcmeNXIpTi8gmSYzKGoK6z+7ODhJAz4C/FmjqI1IN3GOZBIndBXvwMtEvzIc0ud9lnu/1eQH4LJSuNOqkXQRWRBO8AJFh3a4uwxc16U/1aWob3Isz3NfiLTbLpzcsxkQ7/+24Sc2oOSmZ96fCodRJQwuUFoJu/9sB80ybSUOTvxahxbjFXBP+CrJg7FHCQrUTVy+3+ut2A+KsiPFT3SxP0rVtJzCpcm8jBuK9OFAskLoIt1AJTIrlTPuDAMcBHDCSU4qYpO3mWb84PyS26phPz4LPEoHNUv2qb6SODPN9fn29Z+pI7WZVp7Z+KkSLZ0MZhnf "text/plain": [ "<Figure size 640x480 with 2 Axes>" ] }, "metadata": {}, "output_type": "display_data" } ], "source": [ "# imageAnalyser.center = (960, 1040)\n", "# imageAnalyser.span = (100, 100)\n", "# imageAnalyser.fraction = (0.1, 0.1)\n", "\n", "imageAnalyser.center = (960, 875)\n", "imageAnalyser.span = (300, 300)\n", "imageAnalyser.fraction = (0.1, 0.1)\n", "\n", "dataSet_cropOD = imageAnalyser.crop_image(dataSet.OD)\n", "dataSet_cropOD = imageAnalyser.substract_offset(dataSet_cropOD).load()\n", "\n", "dataSet_cropOD.plot.pcolormesh(cmap='jet', vmin=0, col=scanAxis[0], row=scanAxis[1])\n", "plt.show()" ] }, { "attachments": {}, "cell_type": "markdown", "metadata": {}, "source": [ "## Do a 2D two-peak gaussian fit to the OD images" ] }, { "attachments": {}, "cell_type": "markdown", "metadata": {}, "source": [ "### Do the fit" ] }, { "cell_type": "code", "execution_count": 77, "metadata": {}, "outputs": [ { "name": "stderr", "output_type": "stream", "text": [ "f:\\Jianshun\\analyseScript\\Analyser\\FitAnalyser.py:86: RuntimeWarning: invalid value encountered in power\n", " res = (1- ((x-centerx)/(sigmax))**2 - ((y-centery)/(sigmay))**2)**(3 / 2)\n" ] } ], "source": [ "from Analyser.FitAnalyser import ThomasFermi2dModel, DensityProfileBEC2dModel, polylog2_2d\n", "\n", "fitModel = DensityProfileBEC2dModel()\n", "# fitModel = ThomasFermi2dModel()\n", "\n", "fitAnalyser = FitAnalyser(fitModel, fitDim=2)\n", "\n", "# fitAnalyser = FitAnalyser(\"Gaussian-2D\", fitDim=2)\n", "\n", "# dataSet_cropOD = dataSet_cropOD.chunk((1,1,100,100))\n", "\n", "params = fitAnalyser.guess(dataSet_cropOD, guess_kwargs=dict(pureBECThreshold=0.3), dask=\"parallelized\")\n", "fitResult = fitAnalyser.fit(dataSet_cropOD, params).load()" ] }, { "cell_type": "code", "execution_count": 78, "metadata": {}, "outputs": [ { "data": { "text/html": [ "<table><tr><th> name </th><th> value </th><th> initial value </th><th> min </th><th> max </th><th> vary </th><th> expression </th></tr><tr><td> BEC_amplitude </td><td> 366.888620 </td><td> None </td><td> 0.00000000 </td><td> inf </td><td> True </td><td> </td></tr><tr><td> thermal_amplitude </td><td> 0.00000000 </td><td> None </td><td> 0.00000000 </td><td> inf </td><td> True </td><td> </td></tr><tr><td> BEC_centerx </td><td> 152.087577 </td><td> None </td><td> -inf </td><td> inf </td><td> True </td><td> </td></tr><tr><td> BEC_centery </td><td> 156.309927 </td><td> None </td><td> -inf </td><td> inf </td><td> True </td><td> </td></tr><tr><td> thermal_centerx </td><td> 169.884333 </td><td> None </td><td> -inf </td><td> inf </td><td> True </td><td> </td></tr><tr><td> thermal_centery </td><td> 157.547034 </td><td> None </td><td> -inf </td><td> inf </td><td> True </td><td> </td></tr><tr><td> BEC_sigmax </td><td> 1.63603577 </td><td> None </td><td> 0.00000000 </td><td> inf </td><td> True </td><td> </td></tr><tr><td> BEC_sigmay </td><td> 3.42894901 </td><td> None </td><td> 0.00000000 </td><td> inf </td><td> True </td><td> </td></tr><tr><td> thermal_sigmax </td><td> 158.415970 </td><td> None </td><td> 0.00000000 </td><td> inf </td><td> True </td><td> </td></tr><tr><td> thermal_sigmay </td><td> 190.099163 </td><td> None </td><td> -inf </td><td> inf </td><td> False </td><td> thermalAspectRatio * thermal_sigmax </td></tr><tr><td> thermalAspectRatio </td><td> 1.20000000 </td><td> None </td><td> 0.80000000 </td><td> 1.20000000 </td><td> True </td><td> </td></tr><tr><td> condensate_fraction </td><td> 1.00000000 </td><td> None </td><td> -inf </td><td> inf </td><td> False </td><td> BEC_amplitude / (BEC_amplitude + thermal_amplitude) </td></tr></table>" ], "text/plain": [ "Parameters([('BEC_amplitude', <Parameter 'BEC_amplitude', value=366.8886196688306, bounds=[0:inf]>), ('thermal_amplitude', <Parameter 'thermal_amplitude', value=0, bounds=[0:inf]>), ('BEC_centerx', <Parameter 'BEC_centerx', value=152.08757708603804, bounds=[-inf:inf]>), ('BEC_centery', <Parameter 'BEC_centery', value=156.30992724959418, bounds=[-inf:inf]>), ('thermal_centerx', <Parameter 'thermal_centerx', value=169.88433327212883, bounds=[-inf:inf]>), ('thermal_centery', <Parameter 'thermal_centery', value=157.5470340720574, bounds=[-inf:inf]>), ('BEC_sigmax', <Parameter 'BEC_sigmax', value=1.636035766171183, bounds=[0:inf]>), ('BEC_sigmay', <Parameter 'BEC_sigmay', value=3.4289490138791647, bounds=[0:inf]>), ('thermal_sigmax', <Parameter 'thermal_sigmax', value=158.41596952074627, bounds=[0:inf]>), ('thermal_sigmay', <Parameter 'thermal_sigmay', value=190.09916342489552, bounds=[-inf:inf], expr='thermalAspectRatio * thermal_sigmax'>), ('thermalAspectRatio', <Parameter 'thermalAspectRatio', value=1.2, bounds=[0.8:1.2]>), ('condensate_fraction', <Parameter 'condensate_fraction', value=1.0, bounds=[-inf:inf], expr='BEC_amplitude / (BEC_amplitude + thermal_amplitude)'>)])" ] }, "execution_count": 78, "metadata": {}, "output_type": "execute_result" } ], "source": [ "params.compute().item()" ] }, { "cell_type": "code", "execution_count": 79, "metadata": {}, "outputs": [ { "data": { "text/plain": [ "<matplotlib.collections.QuadMesh at 0x1e0e84006d0>" ] }, "execution_count": 79, "metadata": {}, "output_type": "execute_result" }, { "data": { "image/png": "iVBORw0KGgoAAAANSUhEUgAAAnIAAAHECAYAAAC9VcdgAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjUuMSwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy/YYfK9AAAACXBIWXMAAA9hAAAPYQGoP6dpAABPbklEQVR4nO3deXxUVZ738W9lobJUEoILkYgkjhJAIgKiIsQRQRaV1mkGQRFBbKQdm6cbDY50KwRGGxHQ57EVl2aTlm4UUZumWaIooziCBEHWAYUEMCzdBqgsRbbKff6oVFnxZq+kkguf9+t1X97KOefWqcrV/Pyde86xGYZhCAAAAJYT0tIdAAAAQOMQyAEAAFgUgRwAAIBFEcgBAABYFIEcAACARRHIAQAAWBSBHAAAgEWFtXQHLnQVFRU6fvy4YmJiZLPZWro7AIBWzDAMFRQUqEOHDgoJab5cTHFxsUpLSwO+Tps2bRQREdEEPUJNCORa2PHjx9WxY8eW7gYAwEKOHTumyy+/vFmuXVxcrEsiI1XYBNdKSEhQdnY2wVwzIpBrYTExMZVnUyTZW7IrAIBWr0TSS35/O5peaWmpChX4X6USSS+dPKnS0lICuWZEINfCfhxOtUviRgcA1C0Yj+JEK7C/SgQYwcH3DAAATMIrj8ZyN1VHUCsCOQAAYBKmwIIEAozgYPkRAAAAiyJgBgAAJmEKbGi1vKk6gloRyAEAABOGVq2BoVUAAACLImAGAAAmgc5aZWg1OAjkAACACUOr1sDQKgAAgEURMAMAAJNAZ62WNVVHUCsCOQAAYMLQqjUwtAoAAGBRBMwAAMAk0FmrgbRF/RHIAQAAEwI5ayCQAwAAJjwjZw08IwcAAGBRBMwAAMAk0OVHCDCCg+8ZAACYMLRqDQytAgAAWBQBMwAAMGHWqjUQyAEAABOGVq2BoVUAAACLImAGAAAmzFq1Br5nAABgwtCqNTC0CgAAYFEEzAAAwIRZq9ZAIAcAAEwYWrUGvmcAAGDCZAdr4Bk5AAAAiyJgBgAAJjwjZw0EcgAAwIRn5KyBoVUAAACLImAGAAAmYaFSuC2A9oYkd5N1BzUgkAMAACZhYVIYgVyrx9AqAACARZGRAwAAJuEBDq2GG03XF9SMQA4AAJg0ydAqmh1DqwAAABZFRg4AAJiEh0rhAaR7wiuari+oGYEcAAAwC1Vg43YBDMui/gjkAACAWZgCC+TIyAUFz8gBAABYFBk5AABgRkbOEgjkAACAGYGcJTC0CgAAYFFk5AAAgFmIPDNX0aoRyAEAALMwBRbIsfxIUDC0CgAAYFFk5AAAgBkZOUsgIwcAAMxCm+BopIKCAmVkZCg1NVUOh0NxcXHq06eP5s+fr9LS0sZfuNLJkyf1zDPPqHfv3mrXrp0iIyPVqVMnDR06VM8//7zKysoCfo9gsUwgl5eXpyVLluiBBx5Qt27dFB0dLbvdrssvv1z33HOPPvjggxrbZmRkyGaz1Xl89913tfbh0KFDmjRpkpKTkxUREaFLL71UQ4YM0apVq5r64wIAcEE6cuSIrr32Ws2cOVN79uyRYRgqKSlRVlaW0tPTddNNN+nMmTONvv4777yjlJQUPfvss/r6669VVFQku92uo0ePasOGDZo2bZqKioqa8BM1L8sMrSYkJKi8vNz3OiIiQuHh4crNzVVubq7++te/atiwYXrvvfcUFRVV7TXCw8PVrl27Gt8jLKzmr2Pt2rUaOXKkXC6XJCk2NlZ5eXnKzMxUZmamHnroIS1atEg2G7lkAMB5oAWGVt1ut4YPH66cnBxddtllWrZsmQYNGqSKigqtXLlSEydO1I4dOzRmzBitXbu2wddfuXKl7r//flVUVGjUqFF66qmndN1110mSCgsLtXPnTr3//vsKDw9veOdbiGUycuXl5brhhhu0YMECHTp0SOfOnVNhYaGys7P18MMPS5LWrVunSZMm1XiNm2++WSdPnqzxSEpKqrZddna27r33XrlcLvXr108HDhyQ0+mU0+nU9OnTJUlLlizR3Llzm/xzAwDQIkLlCeYaezQiCFy6dKl2794tSVq1apUGDRokSQoJCdGoUaP0xhtvSPL8vd+4cWODrn3ixAlNmjRJFRUVmjJlilasWOEL4iTJ4XCof//+evHFFxUdHd3wzrcQywRyn3zyibZu3apHH31UV155pe/nSUlJWrhwoS+Ae/vtt3Xs2LEmfe/p06erqKhICQkJWrNmjTp37izJ80ufOXOmHnnkEUnSc889F1C6FwCAVqMFnpF76623JEkDBgxQ3759TeWjR49WcnKyJGnZsmUNuvbLL7+sM2fO6PLLL9fzzz/f8M61UpYJ5AYMGFBruTcrJ0lZWVlN9r5FRUW+Z+AeffRRtW3b1lRn2rRpkqT8/Hx9+OGHTfbeAABcKFwul7744gtJ0rBhw6qtY7PZNHToUElSZmZmg67vDfweeOABtWnTJoCeti6WCeTqEhER4Tt3u91Ndt3Nmzfr3Llzkmq+sZKSktS1a1dJDb+xAABolQIZVvUeDbB//35VVHg2aO3evXuN9bxlJ0+e1OnTp+t17ezsbB0/flyS9K//+q/asWOHRo0apYSEBNntdnXs2FGjR4/Wl19+2bBOtwLnTSC3adMm33lqamq1dfbu3avu3bsrMjJSDodDKSkpvgcna7Jnzx7f+TXXXFNjPe+NtXfv3gb2HACAVqiJArn8/PwqR0lJSbVv5w20JCkxMbHGbvmX+bepzcGDB33nX331lW688Ua9++67cjqdioyM1Pfff6933nlH/fr10+zZs+t1zdbivAjkzp496/vi09LSlJKSUm29H374Qfv371dUVJRKSkp08OBBLVy4UL1799bTTz9dbRvvTRIfH1/jbFjpxxurrpuqpKTEdFMDAHC+6tixo+Li4nxHTYFSQUGB77y2v7f+Zf5tauP//PrMmTPVvn17rV+/XkVFRTp79qz279+vgQMHyjAM/fa3v7XUY1KWD+QqKio0duxYnThxQna7XX/4wx9Mda6++mq98MILOnDggIqLi5WXl6eioiJt2LBBvXv3lmEYeu655zR//nxTW+9NUttN5V9e1001e/bsKjd0x44d6/tRAQAInibKyB07dsy30oPT6fQ9Vx5M3iFb7/nKlSs1ZMgQhYR4wqAuXbror3/9qzp06CDJs/6sVVg+kPv1r3+tNWvWSJIWLFigHj16mOqMGTNGU6dOVefOnX1rw7Rp00aDBw/W5s2b1adPH0meX5zT6WzW/k6bNq3KDd3UM2wBAGgSIQpsxmplhBEbG1vlsNvt1b5dTEyM79y7Zmt1/Mv829TGv17//v110003mepER0frP/7jPyRJ33zzjU6dOlWva7c0Swdy6enpeuWVVyRJL730kiZMmNDga0REROj3v/+9JM9igD9dl8b7y6/tpvIvr+umstvtppsaAIALnTcbJkm5ubk11vMv829TG//n6ryTE6vjX3bkyJF6XbulWTaQe/LJJ31DoXPnztVvfvObRl/Lf62aw4cPVynz3iRnzpypNZjz3lj1vakAAGjVgjxrtWvXrr6hTv+Jhj/lLUtISKh1tyZ/3bp1U2ioZ2G72nZgMgzDd26VnZosGchNnTrVt4vCCy+8oPT09GZ7L/8p0LXNSPXeWLXNbAUAwDKCHMhFRUWpX79+kqT169dXW8cwDG3YsEGSNHjw4HpfOyIiQrfccoskad++fTXW279/vyRPEFfTbk+tjeUCufT0dM2bN0+SJ4ibOnVqwNfcsmWL79y7YrRX//79FRkZKanmG+vIkSO+X35DbiwAAPCjcePGSZI+/fRTbd261VS+cuVK38jZgw8+2KBrP/TQQ5I868NWt16cy+XSa6+9Jkm68cYbdckllzTo+i3FUoFcenq6bzh13rx59Qri/NOk1SkpKdHvfvc7SZ4HHQcOHFilPDo6WiNGjJAkvfbaa9VOhpgzZ44kz/Nx99xzT519AgCg1WuBLbrGjRun1NRUGYahESNG+J5b9840nThxoiTPAv0//XudkZEhm80mm82mnJwc07XHjBmjG264QZI0atQobdiwwTeb9X/ "text/plain": [ "<Figure size 640x480 with 2 Axes>" ] }, "metadata": {}, "output_type": "display_data" } ], "source": [ "fitCurve = fitAnalyser.eval(fitResult, x=np.arange(300), y=np.arange(300), dask=\"parallelized\").load()\n", "\n", "fitCurve.plot.pcolormesh(cmap='jet', vmin=0, col=scanAxis[0], row=scanAxis[1])" ] }, { "cell_type": "code", "execution_count": 80, "metadata": {}, "outputs": [], "source": [ "fitModel2 = Polylog22dModel(prefix='thermal_')\n", "fitAnalyser2 = FitAnalyser(fitModel2, fitDim=2)\n", "fitCurve2 = fitAnalyser2.eval(fitResult, x=np.arange(100), y=np.arange(100), dask=\"parallelized\").load()\n", "\n", "fitModel3 = ThomasFermi2dModel(prefix='BEC_')\n", "fitAnalyser3 = FitAnalyser(fitModel3, fitDim=2)\n", "fitCurve3 = fitAnalyser3.eval(fitResult, x=np.arange(100), y=np.arange(100), dask=\"parallelized\").load()" ] }, { "cell_type": "code", "execution_count": 55, "metadata": {}, "outputs": [ { "data": { "image/png": "iVBORw0KGgoAAAANSUhEUgAAAk0AAAHECAYAAAAkrR7VAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjUuMSwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy/YYfK9AAAACXBIWXMAAA9hAAAPYQGoP6dpAACiGUlEQVR4nOzdd3hb5fXA8e/VtGR527Gz916MEGbYBMKmlFXKKBQoq/1RoIUWKJS2tJQCbRllFUqhUDaUGQgzjEzIhuwd76Vh7fv74+peXcnySjxi+3yeJ09l6Uq+ppZ87jnnPa+iqqqKEEIIIYRolaWnT0AIIYQQojeQoEkIIYQQoh0kaBJCCCGEaAcJmoQQQggh2kGCJiGEEEKIdpCgSQghhBCiHSRoEkIIIYRoB1tPn0BvF4/H2blzJzk5OSiK0tOnI4QQQoh2UFUVr9fLoEGDsFjal0OSoGkP7dy5k6FDh/b0aQghhBBiN2zbto0hQ4a061gJmvZQTk4OoP1Hz83N7eGzEUIIIUR7NDY2MnToUOPveHtI0LSH9JJcbm6uBE1CCCFEL9OR1hppBBdCCCGEaAcJmoQQQggh2kGCJiGEEEKIdpCgSQghhBCiHSRoEkIIIYRoBwmahBBCCCHaQYImIYQQQoh2kKBJCCGEEKIdJGgSQgghhGgHCZqEEEIIIdpBgiYhhBBCiHaQoEkIIYQQoh0kaBJC9GuqqtIUjvX0aQghegEJmoQQ/dqvXl3BvnfOZVttoKdPRQixl5OgSQjRry3eXEcwEmf1rsaePhUhxF5OgiYhRL8WSJTmfMFoD5+JEGJvJ0GTEKJf84e1YMkXkqBJCNE6CZqEEP1aIJTINEnQJIRogwRNQoh+KxyNE47FAQmahBBtk6BJCNHvVDYG+WpjTcqoAelpEkK0RYImIUS/87Pnv+HcR79i0eZa4z6/ZJqEEG2QoEkI0e9srPYB8F2F17jPK0GTEKINEjQJIfoVVVWp9YcBKG8IGvdLpkkI0RYJmoQQ/UpjMEokpgJQ3pgMmqQRXAjRFgmahBD9So0vZNyuMAdNu9kIrqrqHp+TEKJ3kKBJCNGv6KU5SC3P7U6m6fHPNjLzD/PYUOXrlHMTQuzdJGgSQvQr1b5k0FRlyjrtTtD04beVVHlDLDatwhNC9F0SNAkh+pUafzJQMlfWAuEYsXjHSm2hqDYY0zzvSQjRd0nQJIToV2pNmaZ0+j507RWKasFSICJBkxD9gQRNQoh+pcbfctDU0WbwUEQyTUL0JxI0CSH6lWpTH1O6jvY16eW5gARNQvQLEjQJIfqV2tYyTR0OmhLlOQmahOgXJGgSQvQrNa30NHW4PGc0gstgTCH6g14TNC1dupQ77riDU089lQkTJlBUVITdbqeoqIhDDz2U3//+99TWtr7st6Kiguuvv57x48fjcrkoLCxk1qxZPP744zKgToh+wrx6Ll2HM00RKc8J0Z/YevoE2uuf//wnDz74oPF1VlYWLpeL2tpavvjiC7744gvuv/9+3njjDQ4++OBmz1+yZAnHH388NTU1AHg8HrxeL/Pnz2f+/Pm8+OKLvPHGGzidzm77mYQQ3SseVzutPKeqqlGea5LVc0L0C70m0zRz5kz+/Oc/8+WXX1JXV0dTUxONjY14vV6eeuopSkpKqK6u5vTTT6ehoSHluQ0NDZx88snU1NQwYcIEFi1ahNfrxe/388ADD2C325k7dy7XXXddD/10QojuUN8UQR/F5LJbjfvzXHagY+W5aFw1XksyTUL0D70maLrwwgu54YYbOOigg8jPzzfu93g8XHTRRTzzzDMAVFZW8uabb6Y895577qG8vByXy8Xbb7/NjBkzAHA4HFx99dXccccdADz66KOsXbu2e34gIUS30/edy3PZKXDbjftLc7UMc0cyTXo/E0jQJER/0WuCprYcdNBBxu3t27enPPb0008DcO655zJy5Mhmz7322mvxeDzEYjGeffbZrj1RIUSP0Wc0FWU7yMkyB01ZAPg7EjSZSnJBKc8J0S/0maDps88+M26PHj3auP3dd9+xdetWAObMmZPxuR6Ph1mzZgEwd+7cLjxLIURPqvJqmaYij4OcrGRLZ0mOlmny7namSVbPCdEf9OqgKRQKsXnzZh544AEuuOACAMaMGcMpp5xiHLNy5Urj9pQpU1p8Lf2x1atXt/k9GxsbU/4JIXqHHfVNAAzOd5Hr2sNMk5TnhOh3es3qObOsrCxCoebLhg899FD+85//pKyA27lzp3F78ODBLb6m/lhjYyM+nw+Px5PxuLvuusvogRJC9C7b6wIADClwsy1xG6A0kWnqSCO4vnIOZBsVIfqLXplpKisro7S0lOzsbOO+o446ivvvv59hw4alHOv1eo3bbre7xdc0P2Z+Trqbb76ZhoYG49+2bdt250cQQvSA7XVapmlIgcsoz1kUKN6N8lwwksw0ReMqYVPmSQjRN/XKoGnz5s2Ul5fj8/moqKjgnnvu4ZtvvmHmzJncdtttXfq9nU4nubm5Kf+EEL1DMmhyG43g2Q6bMXKgIRBp92uF0pq/JdskRN/XK4MmswEDBnD99dfz7rvvoigKd955Z8rIgZycHON2IBDI9BLNHjM/RwjRN6iqairPJTNNbqeVATlaT1OlN9ju1wulZZYCEWkGF6Kv6/VBk27mzJkcdthhgDZvSTdo0CDj9o4dO1p8vv5Ybm5ui/1MQojeq8YfJhiJoygwMD+LXFOmSZ/TVBeIpPQqtaZZ0CSZJiH6vD4TNEGymXv9+vXGfeYVc+aVdOn0xyZNmtRFZyeE6El6aa40JwunzZqSacpz2XHYtI/DysaW96YzSw+uOqM89/DHG3hpyfa2DxRC9Ig+FTRt3LgRSC2vjR8/3mgOf/fddzM+z+/3G3OeZs+e3cVnKYToCebSHMDkQXnYrQpTB+ejKAoDEs3gmUp0W2r8vLRke0qzdyjSuZmmHfVN/Ondb7n1tZYv7oQQPatXBE2xWAxVVVs9Zt68eSxcuBCAI488MuWxCy+8EIDnn3+ezZs3N3vugw8+iM/nw2q1cv7553fKOQsh9i7mlXMAYwZ4WHLrcfzhDC0brc9qypRp+s0bq7jhxWVc+q9FxmdRenluTzftrWwMGq8jK/GE2Dv1iqBp27Zt7LvvvjzyyCNs3LgxJYDatm0bf/zjHznttNNQVZXCwsJmG+/ecMMNlJWVEQgEOOmkk1iyZAkA4XCYhx9+mFtvvRWAyy+/nHHjxnXfDyaE6DbmGU263Cw7iqIAyf3nKhqbZ5o+/q4KgM/WVfPIp1pGu3l5bs8awWt8YeN2R4ZsCiG6T68Zbrls2TJ+8pOfANpGu7m5uTQ1NeH3+41jRo4cycsvv0xZWVnKc/Py8njzzTc5/vjjWb16NTNmzCAnJ4dgMEgkoi0xnj17Nvfdd1/3/UBCiG6VnmlKp6+gq0hstVLlDfFteSOHjSmmwG2nLjGO4L+LtvGTI0Z3eiN4jT+Z4fKHoxRkO/bo9YQQna9XZJoGDRrECy+8wFVXXcX+++9PcXExjY2NxONxhg0bximnnMLjjz/OqlWr2HfffTO+xv7778+qVau47rrrGDt2LJFIhOzsbA477DAee+wx3nnnnZRJ4kKIvmVnYguVQfktBE2JTJNenvvly8u54ImFfLmhhvqm5Pymap/2eGf3NOmbCQP4Q7IST4i9Ua/INDkcDs466yzOOuusPXqd0tJS7r33Xu69995OOjMhRG9RmwhKij2ZL45K02Y1fbtL21dy0eY6zC2V3mCUcDTe6avnUspzsgGwEHulXpFpEkKIPaGqqlFeK2yh7KU3glc0BonE4pQnepu+LdeCpzyXHYvW/kR9INz55TlfsjwXkEyTEHslCZqEEH1eYzBKLK6li/Ld9ozHJBvBQ5Q3BEkczppExqnY46DArQVcNf5ws0yTeSL4go01/Py/36QEQm0xl+d80gguxF5JgiYhRJ9XlwhI3A4rWXZrxmP0RvCGpgjrq3zG/VtqtVV3RdlOI0tV5w8bPU12q5Z+MpfnHvtsE698vYN3Vpa3+xzN5bmAlOeE2CtJ0CSE6PPqAlpAome "text/plain": [ "<Figure size 640x480 with 1 Axes>" ] }, "metadata": {}, "output_type": "display_data" }, { "data": { "image/png": "iVBORw0KGgoAAAANSUhEUgAAAk0AAAHECAYAAAAkrR7VAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjUuMSwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy/YYfK9AAAACXBIWXMAAA9hAAAPYQGoP6dpAACfOklEQVR4nOzdd5hcZdn48e+ZPrO9ZpNNNr03QiAEQiAQCIQAAgqCFBUVVORVBDso6OuLAqL+RLCgItJ7CQSC1CSQ3nvdlO116u7U8/vjzDkzsyXZTTZb78917cXszDkzz4ads/fcz/3cj6KqqooQQgghhDgqU08PQAghhBCiL5CgSQghhBCiAyRoEkIIIYToAAmahBBCCCE6QIImIYQQQogOkKBJCCGEEKIDJGgSQgghhOgAS08PoK+LxWKUl5eTkZGBoig9PRwhhBBCdICqqni9XoYMGYLJ1LEckgRNJ6i8vJxhw4b19DCEEEIIcRwOHz7M0KFDO3SsBE0nKCMjA9D+0TMzM3t4NEIIIYToCI/Hw7Bhw4y/4x0hQdMJ0qfkMjMzJWgSQggh+pjOlNZIIbgQQgghRAdI0CSEEEII0QESNAkhhBBCdIAETUIIIYQQHSBBkxBCCCFEB0jQJIQQQgjRARI0CSGEEEJ0gARNQgghhBAdIEGTEEIIIUQHSNAkhBBCCNEBEjQJIYQQQnSABE1CCCGEEB0gG/YKIYSqQqAO1Bg4ssBi7+kRCSF6IQmahBADUywKO96Ezc9D6XIIerT7TRYYMgMmXganfhmc2T06TCFE7yFBkxBi4DmwDJb8CKq3Jd2pACrEInBkjfb18QNw/t0w6xYwmXtqtEKIXkKCJiHEwBGLwof/B8t+B6jaVNxpX4PJV0DBBDBZwX0Y9r0Pq/8O1dvhnR/D9jfgmichvaCnfwIhRA9SVFVVe3oQfZnH4yErKwu3201mZmZPD0cI0Z5IEF7+Oux4Q/v+1JvggvvAldv28bEYrH8Clv4cQl7IGgY3vAIF47ptyEKIk+d4/n7L6jkhRP8XCcKz12oBk9kGVz0Ol/+p/YAJwGSC026Gb7wPuaO1DNS/L4WaXd03biFEryJBkxCif4tF4ZVbYN8HYE2D61+CaVd3/PyC8fC1pTBoCviq4MkrwFMOQFMoyrZyN5KwF2JgkKBJCNG/ffAr2P6aVq907VMw6tzOP0daPtz0BuSPB285PHMNBH38/PWtLPp/y/lsX12XD1sI0ftI0CSE6L+2vQbLf6/dvuIxGH3+8T9XWh5c/wKkFUDlFlh8Bzsq3ADsr/Wf+FiFEL2eBE1CiP6poRRe/452+6zbOzcl156cEXDNf0Axw5YXOMu9GAB/MHLizy2E6PUkaBJC9D/RCLz8DW3VW8mZMP/ernvu4WfC/J8DcEfkX4xQKiRoEmKAkKBJCNH/fPYnOLIa7Jlw1d/A3MUt6c76H0Ilc3EqIX5n/Qv+5lDXPr8QoleSoEkI0b/U7oUP79duL/wtZJd0/WuYTJSe/QBe1clM0x6mVbzY9a8hhOh1JGgSQvQfqgqLvwfRoFb0Pf26LnnaBn+I9YcaUu4rU/P5TUR7/gVVj4OvukteSwjRe0nQJIToP7a/DqXLwOKES/8AitIlT3vP61u56tFP+Xh3jXFfjTfIs9Hz2RIbgTPmh//e2yWvJYTovSRoEkL0D+FmeO8e7fac/4Gc4V321NvLPQC8u63SuK/GGySGiZ+Hv6rdsfFpOLy6y15TCNH7SNAkhOgfVj4KjYcgYwjM+W6XPa2qqlS4mwH4ZHeN0f272qPdt0Edy1LbBdrBb92pdSAXQvRLEjQJIfo+byUs+512+4J7wZbWZU/taY7QFNYCoSMNTZTWBQCo8QWNYx4x3wCOLKjcDBue6rLXFkL0LhI0CSH6vg9+BSEfFM+EqV3QxDJJVTyjpPskXtdU7UkETWWhdDjnh9o3Hz+gbRAshOh3JGgSQvRttXth4zPa7YvuB1PXXtYq3e0ETd5EYOQLRuD0r0HGYPAcgXVPdOkYhBC9Q58JmtavX899993H5ZdfzoQJE8jLy8NqtZKXl8ecOXP49a9/TX19/VGfo6qqijvvvJPx48fjdDrJzc1l7ty5PP7447JLuRB91ScPgBqDcRdDyRmdPn3F3lruX7KDSDTW5uN60JTh0Bpk7q72oqoq1d5EMBWMxAib7HDOD+JjeghCsh+dEP1Nnwma/vnPf3Lvvffy5ptvsmvXLgKBAE6nk/r6ej799FPuvvtuxo8fz2effdbm+evWrWPy5Mk8/PDD7N69G4vFgtfrZfny5XzjG9/g4osvJhiUlLoQfUrtHtgSbyw578edPj0SjXH946v468f7+e+OqjaPqYxPz00sygSg0R/GF4zQHE4NsvzBCMy4EbKHg78aVv+90+MRQvRufSZomjVrFg8++CCfffYZDQ0NNDU14fF48Hq9PPHEExQUFFBbW8sVV1yB2+1OOdftdnPppZdSV1fHhAkTWLNmDV6vF7/fzyOPPILVamXp0qXccccdPfTTCSGOy8d6lmkhDJnR6dOX7ak1bgcj7WSa4kHThMEZAHiDEWM1Xbrdgs2iXUZ9wQhYbIngbcUfoNnd6vmEEH1XnwmabrrpJu666y5mz55Ndna2cX96ejpf/vKXeeopbcVKdXU1ixcvTjn3oYceorKyEqfTydtvv81pp50GgM1m47bbbuO+++4D4G9/+xu7d+/unh9ICHFianbD1pe028eRZQJ4ef0R43Z7QVNVPEAaNyjDuG9PlQ+Awgw76XZt2s4fjLcamPZFyB8HTQ2SbRKin+kzQdOxzJ4927h95MiRlMeefPJJAK699lpGjhzZ6tzbb7+d9PR0otEoTz/99MkdqBCia+i1TOMvgSGndPp0T3OYpdsTU3KBYCTl8bWl9fxu6S4O1mstBoqznWTG65r2VmtBU366nTS7GYhnmgBMZph7l3Z71V8g3NTpsQkheqd+EzQtW7bMuD169Gjj9q5duzh06BAACxcubPPc9PR05s6dC8DSpUtP4iiFEF2ioRS2vqzdPvdHKQ95m8NUuI8dqHy2r45QUnYpEE40pVRVlbte3MSfPthrBEiDMh3kpNkA2Fuj3ZeXbiPNpmeakoKuKVdBVgn4a7RO4UKIfqFPB03BYJDS0lIeeeQRbrzxRgDGjBnDZZddZhyzdetW4/aUKVPafS79se3btx/zNT0eT8qXEKKbrXxMyzKNPj8ly6SqKgv/uIxzHviwVauAA7V+3t1WaayUPdKQGlgFglE+3FXN3a9tYU1pg9HEUleU5SDbpQVNe6q8gBY0JabnkoImsxXO+o52+9M/QTQ1iyWE6JssPT2A4+FwONpc6TZnzhyeeeYZ7Ha7cV95eblxu7i4uN3n1B/zeDz4fD7S09PbPO7+++83aqCEED0gUA/r/6PdPuv2lIfWH2o0gqFdVV6KshzGY//z7Aa2lLlZfPvZTCnOoqxl0BSK8p2n1+MPRXlq5aFWL5vjspLjsgKwv1ZrJ6BNz2nP42sxvceMG+Cj32hZsR2vw5TPH/ePLIToHfpkpqmoqIhBgwaRlpbYKuG8887jD3/4AyUlJSnHer1e47bL5Wr3OZMfSz6npZ/85Ce43W7j6/Dhw8fzIwghjte6f0HYD4OmwqjzUh56YU3i/RhL6r0WjsbYUaFlhQ/GM0jljVqwk5+ufcgKhCL4Q6n7xplNinFbURRy4pkmfVovL93edqYJtK1czrhVu738DyC94ITo8/pk0FRaWkplZSU+n4+qqioeeughNm7cyKxZs/j5z39+Ul/bbreTmZmZ8iWE6CaRIKz6q3b7rNsJRmM8veogRxoC+IMRFm9OZJabkwKgg3V+IjEtaKnza1nq8njd09hCLavsD0XJT7cZ51jNCh/ceS5zx+Zz3+WTAciOZ5p0+Wk2oxC8ZcAFwKxbwOrS9qTb/+GJ/ORCiF6gTwZNyQoLC7nzzjt55513UBSFX/3qVyktBzIyEsuEA4FAW0/R6rHkc4QQvciWF8FXBRlDYMpVvLutip+9upUH3tnFR7tqUgKX5kjitt4iAKDWFwISmaYx8aCpKRRJmWK "text/plain": [ "<Figure size 640x480 with 1 Axes>" ] }, "metadata": {}, "output_type": "display_data" } ], "source": [ "fig = plt.figure()\n", "ax = fig.gca()\n", "\n", "dataSet_cropOD.sum(dim='x').plot(ax=ax, col=scanAxis[0], row=scanAxis[1])\n", "fitCurve.sum(dim='x').plot(ax=ax, col=scanAxis[0], row=scanAxis[1])\n", "fitCurve2.sum(dim='x').plot(ax=ax, col=scanAxis[0], row=scanAxis[1])\n", "fitCurve3.sum(dim='x').plot(ax=ax, col=scanAxis[0], row=scanAxis[1])\n", "\n", "plt.show()\n", "\n", "fig = plt.figure()\n", "ax = fig.gca()\n", "\n", "dataSet_cropOD.sum(dim='y').plot(ax=ax, col=scanAxis[0], row=scanAxis[1])\n", "fitCurve.sum(dim='y').plot(ax=ax, col=scanAxis[0], row=scanAxis[1])\n", "fitCurve2.sum(dim='y').plot(ax=ax, col=scanAxis[0], row=scanAxis[1])\n", "fitCurve3.sum(dim='y').plot(ax=ax, col=scanAxis[0], row=scanAxis[1])\n", "\n", "plt.show()" ] }, { "cell_type": "code", "execution_count": 56, "metadata": {}, "outputs": [ { "name": "stderr", "output_type": "stream", "text": [ "C:\\Users\\data\\AppData\\Roaming\\Python\\Python39\\site-packages\\numpy\\lib\\function_base.py:2246: RuntimeWarning: invalid value encountered in _get_fit_full_result_single (vectorized)\n", " outputs = ufunc(*inputs)\n" ] } ], "source": [ "value = fitAnalyser.get_fit_full_result(fitResult)" ] }, { "cell_type": "code", "execution_count": 57, "metadata": {}, "outputs": [ { "data": { "text/html": [ "<div><svg style=\"position: absolute; width: 0; height: 0; overflow: hidden\">\n", "<defs>\n", "<symbol id=\"icon-database\" viewBox=\"0 0 32 32\">\n", "<path d=\"M16 0c-8.837 0-16 2.239-16 5v4c0 2.761 7.163 5 16 5s16-2.239 16-5v-4c0-2.761-7.163-5-16-5z\"></path>\n", "<path d=\"M16 17c-8.837 0-16-2.239-16-5v6c0 2.761 7.163 5 16 5s16-2.239 16-5v-6c0 2.761-7.163 5-16 5z\"></path>\n", "<path d=\"M16 26c-8.837 0-16-2.239-16-5v6c0 2.761 7.163 5 16 5s16-2.239 16-5v-6c0 2.761-7.163 5-16 5z\"></path>\n", "</symbol>\n", "<symbol id=\"icon-file-text2\" viewBox=\"0 0 32 32\">\n", "<path d=\"M28.681 7.159c-0.694-0.947-1.662-2.053-2.724-3.116s-2.169-2.030-3.116-2.724c-1.612-1.182-2.393-1.319-2.841-1.319h-15.5c-1.378 0-2.5 1.121-2.5 2.5v27c0 1.378 1.122 2.5 2.5 2.5h23c1.378 0 2.5-1.122 2.5-2.5v-19.5c0-0.448-0.137-1.23-1.319-2.841zM24.543 5.457c0.959 0.959 1.712 1.825 2.268 2.543h-4.811v-4.811c0.718 0.556 1.584 1.309 2.543 2.268zM28 29.5c0 0.271-0.229 0.5-0.5 0.5h-23c-0.271 0-0.5-0.229-0.5-0.5v-27c0-0.271 0.229-0.5 0.5-0.5 0 0 15.499-0 15.5 0v7c0 0.552 0.448 1 1 1h7v19.5z\"></path>\n", "<path d=\"M23 26h-14c-0.552 0-1-0.448-1-1s0.448-1 1-1h14c0.552 0 1 0.448 1 1s-0.448 1-1 1z\"></path>\n", "<path d=\"M23 22h-14c-0.552 0-1-0.448-1-1s0.448-1 1-1h14c0.552 0 1 0.448 1 1s-0.448 1-1 1z\"></path>\n", "<path d=\"M23 18h-14c-0.552 0-1-0.448-1-1s0.448-1 1-1h14c0.552 0 1 0.448 1 1s-0.448 1-1 1z\"></path>\n", "</symbol>\n", "</defs>\n", "</svg>\n", "<style>/* CSS stylesheet for displaying xarray objects in jupyterlab.\n", " *\n", " */\n", "\n", ":root {\n", " --xr-font-color0: var(--jp-content-font-color0, rgba(0, 0, 0, 1));\n", " --xr-font-color2: var(--jp-content-font-color2, rgba(0, 0, 0, 0.54));\n", " --xr-font-color3: var(--jp-content-font-color3, rgba(0, 0, 0, 0.38));\n", " --xr-border-color: var(--jp-border-color2, #e0e0e0);\n", " --xr-disabled-color: var(--jp-layout-color3, #bdbdbd);\n", " --xr-background-color: var(--jp-layout-color0, white);\n", " --xr-background-color-row-even: var(--jp-layout-color1, white);\n", " --xr-background-color-row-odd: var(--jp-layout-color2, #eeeeee);\n", "}\n", "\n", "html[theme=dark],\n", "body[data-theme=dark],\n", "body.vscode-dark {\n", " --xr-font-color0: rgba(255, 255, 255, 1);\n", " --xr-font-color2: rgba(255, 255, 255, 0.54);\n", " --xr-font-color3: rgba(255, 255, 255, 0.38);\n", " --xr-border-color: #1F1F1F;\n", " --xr-disabled-color: #515151;\n", " --xr-background-color: #111111;\n", " --xr-background-color-row-even: #111111;\n", " --xr-background-color-row-odd: #313131;\n", "}\n", "\n", ".xr-wrap {\n", " display: block !important;\n", " min-width: 300px;\n", " max-width: 700px;\n", "}\n", "\n", ".xr-text-repr-fallback {\n", " /* fallback to plain text repr when CSS is not injected (untrusted notebook) */\n", " display: none;\n", "}\n", "\n", ".xr-header {\n", " padding-top: 6px;\n", " padding-bottom: 6px;\n", " margin-bottom: 4px;\n", " border-bottom: solid 1px var(--xr-border-color);\n", "}\n", "\n", ".xr-header > div,\n", ".xr-header > ul {\n", " display: inline;\n", " margin-top: 0;\n", " margin-bottom: 0;\n", "}\n", "\n", ".xr-obj-type,\n", ".xr-array-name {\n", " margin-left: 2px;\n", " margin-right: 10px;\n", "}\n", "\n", ".xr-obj-type {\n", " color: var(--xr-font-color2);\n", "}\n", "\n", ".xr-sections {\n", " padding-left: 0 !important;\n", " display: grid;\n", " grid-template-columns: 150px auto auto 1fr 20px 20px;\n", "}\n", "\n", ".xr-section-item {\n", " display: contents;\n", "}\n", "\n", ".xr-section-item input {\n", " display: none;\n", "}\n", "\n", ".xr-section-item input + label {\n", " color: var(--xr-disabled-color);\n", "}\n", "\n", ".xr-section-item input:enabled + label {\n", " cursor: pointer;\n", " color: var(--xr-font-color2);\n", "}\n", "\n", ".xr-section-item input:enabled + label:hover {\n", " color: var(--xr-font-color0);\n", "}\n", "\n", ".xr-section-summary {\n", " grid-column: 1;\n", " color: var(--xr-font-color2);\n", " font-weight: 500;\n", "}\n", "\n", ".xr-section-summary > span {\n", " display: inline-block;\n", " padding-left: 0.5em;\n", "}\n", "\n", ".xr-section-summary-in:disabled + label {\n", " color: var(--xr-font-color2);\n", "}\n", "\n", ".xr-section-summary-in + label:before {\n", " display: inline-block;\n", " content: 'â–º';\n", " font-size: 11px;\n", " width: 15px;\n", " text-align: center;\n", "}\n", "\n", ".xr-section-summary-in:disabled + label:before {\n", " color: var(--xr-disabled-color);\n", "}\n", "\n", ".xr-section-summary-in:checked + label:before {\n", " content: 'â–¼';\n", "}\n", "\n", ".xr-section-summary-in:checked + label > span {\n", " display: none;\n", "}\n", "\n", ".xr-section-summary,\n", ".xr-section-inline-details {\n", " padding-top: 4px;\n", " padding-bottom: 4px;\n", "}\n", "\n", ".xr-section-inline-details {\n", " grid-column: 2 / -1;\n", "}\n", "\n", ".xr-section-details {\n", " display: none;\n", " grid-column: 1 / -1;\n", " margin-bottom: 5px;\n", "}\n", "\n", ".xr-section-summary-in:checked ~ .xr-section-details {\n", " display: contents;\n", "}\n", "\n", ".xr-array-wrap {\n", " grid-column: 1 / -1;\n", " display: grid;\n", " grid-template-columns: 20px auto;\n", "}\n", "\n", ".xr-array-wrap > label {\n", " grid-column: 1;\n", " vertical-align: top;\n", "}\n", "\n", ".xr-preview {\n", " color: var(--xr-font-color3);\n", "}\n", "\n", ".xr-array-preview,\n", ".xr-array-data {\n", " padding: 0 5px !important;\n", " grid-column: 2;\n", "}\n", "\n", ".xr-array-data,\n", ".xr-array-in:checked ~ .xr-array-preview {\n", " display: none;\n", "}\n", "\n", ".xr-array-in:checked ~ .xr-array-data,\n", ".xr-array-preview {\n", " display: inline-block;\n", "}\n", "\n", ".xr-dim-list {\n", " display: inline-block !important;\n", " list-style: none;\n", " padding: 0 !important;\n", " margin: 0;\n", "}\n", "\n", ".xr-dim-list li {\n", " display: inline-block;\n", " padding: 0;\n", " margin: 0;\n", "}\n", "\n", ".xr-dim-list:before {\n", " content: '(';\n", "}\n", "\n", ".xr-dim-list:after {\n", " content: ')';\n", "}\n", "\n", ".xr-dim-list li:not(:last-child):after {\n", " content: ',';\n", " padding-right: 5px;\n", "}\n", "\n", ".xr-has-index {\n", " font-weight: bold;\n", "}\n", "\n", ".xr-var-list,\n", ".xr-var-item {\n", " display: contents;\n", "}\n", "\n", ".xr-var-item > div,\n", ".xr-var-item label,\n", ".xr-var-item > .xr-var-name span {\n", " background-color: var(--xr-background-color-row-even);\n", " margin-bottom: 0;\n", "}\n", "\n", ".xr-var-item > .xr-var-name:hover span {\n", " padding-right: 5px;\n", "}\n", "\n", ".xr-var-list > li:nth-child(odd) > div,\n", ".xr-var-list > li:nth-child(odd) > label,\n", ".xr-var-list > li:nth-child(odd) > .xr-var-name span {\n", " background-color: var(--xr-background-color-row-odd);\n", "}\n", "\n", ".xr-var-name {\n", " grid-column: 1;\n", "}\n", "\n", ".xr-var-dims {\n", " grid-column: 2;\n", "}\n", "\n", ".xr-var-dtype {\n", " grid-column: 3;\n", " text-align: right;\n", " color: var(--xr-font-color2);\n", "}\n", "\n", ".xr-var-preview {\n", " grid-column: 4;\n", "}\n", "\n", ".xr-index-preview {\n", " grid-column: 2 / 5;\n", " color: var(--xr-font-color2);\n", "}\n", "\n", ".xr-var-name,\n", ".xr-var-dims,\n", ".xr-var-dtype,\n", ".xr-preview,\n", ".xr-attrs dt {\n", " white-space: nowrap;\n", " overflow: hidden;\n", " text-overflow: ellipsis;\n", " padding-right: 10px;\n", "}\n", "\n", ".xr-var-name:hover,\n", ".xr-var-dims:hover,\n", ".xr-var-dtype:hover,\n", ".xr-attrs dt:hover {\n", " overflow: visible;\n", " width: auto;\n", " z-index: 1;\n", "}\n", "\n", ".xr-var-attrs,\n", ".xr-var-data,\n", ".xr-index-data {\n", " display: none;\n", " background-color: var(--xr-background-color) !important;\n", " padding-bottom: 5px !important;\n", "}\n", "\n", ".xr-var-attrs-in:checked ~ .xr-var-attrs,\n", ".xr-var-data-in:checked ~ .xr-var-data,\n", ".xr-index-data-in:checked ~ .xr-index-data {\n", " display: block;\n", "}\n", "\n", ".xr-var-data > table {\n", " float: right;\n", "}\n", "\n", ".xr-var-name span,\n", ".xr-var-data,\n", ".xr-index-name div,\n", ".xr-index-data,\n", ".xr-attrs {\n", " padding-left: 25px !important;\n", "}\n", "\n", ".xr-attrs,\n", ".xr-var-attrs,\n", ".xr-var-data,\n", ".xr-index-data {\n", " grid-column: 1 / -1;\n", "}\n", "\n", "dl.xr-attrs {\n", " padding: 0;\n", " margin: 0;\n", " display: grid;\n", " grid-template-columns: 125px auto;\n", "}\n", "\n", ".xr-attrs dt,\n", ".xr-attrs dd {\n", " padding: 0;\n", " margin: 0;\n", " float: left;\n", " padding-right: 10px;\n", " width: auto;\n", "}\n", "\n", ".xr-attrs dt {\n", " font-weight: normal;\n", " grid-column: 1;\n", "}\n", "\n", ".xr-attrs dt:hover span {\n", " display: inline-block;\n", " background: var(--xr-background-color);\n", " padding-right: 10px;\n", "}\n", "\n", ".xr-attrs dd {\n", " grid-column: 2;\n", " white-space: pre-wrap;\n", " word-break: break-all;\n", "}\n", "\n", ".xr-icon-database,\n", ".xr-icon-file-text2,\n", ".xr-no-icon {\n", " display: inline-block;\n", " vertical-align: middle;\n", " width: 1em;\n", " height: 1.5em !important;\n", " stroke-width: 0;\n", " stroke: currentColor;\n", " fill: currentColor;\n", "}\n", "</style><pre class='xr-text-repr-fallback'><xarray.Dataset>\n", "Dimensions: ()\n", "Data variables:\n", " BEC_amplitude object 0.0+/-nan\n", " thermal_amplitude object 2104.548431645919+/-nan\n", " BEC_centerx object 146.94301032591366+/-nan\n", " BEC_centery object 147.47224593536436+/-nan\n", " thermal_centerx object 146.27287010988167+/-nan\n", " thermal_centery object 148.78153517037947+/-nan\n", " BEC_sigmax object 17.155488681677085+/-nan\n", " BEC_sigmay object 18.315601451967396+/-nan\n", " thermal_sigmax object 42.999686622150065+/-nan\n", " thermal_sigmay object 51.599623946580074+/-nan\n", " thermalAspectRatio object 1.2+/-nan\n", " condensate_fraction object 0.0+/-nan\n", "Attributes:\n", " IMAGE_SUBCLASS: IMAGE_GRAYSCALE\n", " IMAGE_VERSION: 1.2\n", " IMAGE_WHITE_IS_ZERO: 0\n", " x_start: 810\n", " x_end: 1110\n", " y_end: 1025\n", " y_start: 725\n", " x_center: 960\n", " y_center: 875\n", " x_span: 300\n", " y_span: 300</pre><div class='xr-wrap' style='display:none'><div class='xr-header'><div class='xr-obj-type'>xarray.Dataset</div></div><ul class='xr-sections'><li class='xr-section-item'><input id='section-53523a01-2c07-4165-91ae-60ce98257240' class='xr-section-summary-in' type='checkbox' disabled ><label for='section-53523a01-2c07-4165-91ae-60ce98257240' class='xr-section-summary' title='Expand/collapse section'>Dimensions:</label><div class='xr-section-inline-details'></div><div class='xr-section-details'></div></li><li class='xr-section-item'><input id='section-448573c2-6f7b-479e-ae18-5c693d5ebbdb' class='xr-section-summary-in' type='checkbox' disabled ><label for='section-448573c2-6f7b-479e-ae18-5c693d5ebbdb' class='xr-section-summary' title='Expand/collapse section'>Coordinates: <span>(0)</span></label><div class='xr-section-inline-details'></div><div class='xr-section-details'><ul class='xr-var-list'></ul></div></li><li class='xr-section-item'><input id='section-f81fa7c4-b59b-456c-929b-d5fdf0999ba7' class='xr-section-summary-in' type='checkbox' checked><label for='section-f81fa7c4-b59b-456c-929b-d5fdf0999ba7' class='xr-section-summary' >Data variables: <span>(12)</span></label><div class='xr-section-inline-details'></div><div class='xr-section-details'><ul class='xr-var-list'><li class='xr-var-item'><div class='xr-var-name'><span>BEC_amplitude</span></div><div class='xr-var-dims'>()</div><div class='xr-var-dtype'>object</div><div class='xr-var-preview xr-preview'>0.0+/-nan</div><input id='attrs-ae17e55d-accd-49e8-9a73-373506561892' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-ae17e55d-accd-49e8-9a73-373506561892' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-4d309f58-8d37-475d-9462-884f4ad37e77' class='xr-var-data-in' type='checkbox'><label for='data-4d309f58-8d37-475d-9462-884f4ad37e77' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array(0.0+/-nan, dtype=object)</pre></div></li><li class='xr-var-item'><div class='xr-var-name'><span>thermal_amplitude</span></div><div class='xr-var-dims'>()</div><div class='xr-var-dtype'>object</div><div class='xr-var-preview xr-preview'>2104.548431645919+/-nan</div><input id='attrs-ce8230fc-40ed-4545-8712-ec500a361b66' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-ce8230fc-40ed-4545-8712-ec500a361b66' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-c4d45705-76ae-4681-a822-8d9af1dbbece' class='xr-var-data-in' type='checkbox'><label for='data-c4d45705-76ae-4681-a822-8d9af1dbbece' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array(2104.548431645919+/-nan, dtype=object)</pre></div></li><li class='xr-var-item'><div class='xr-var-name'><span>BEC_centerx</span></div><div class='xr-var-dims'>()</div><div class='xr-var-dtype'>object</div><div class='xr-var-preview xr-preview'>146.94301032591366+/-nan</div><input id='attrs-d3679e45-9660-4dc8-9900-6ff9b96700a5' class='xr-var-attrs-in' type='checkbox' disabled><label for='attrs-d3679e45-9660-4dc8-9900-6ff9b96700a5' title='Show/Hide attributes'><svg class='icon xr-icon-file-text2'><use xlink:href='#icon-file-text2'></use></svg></label><input id='data-2d54cdcf-9b44-4720-b151-cdd4cfffd6be' class='xr-var-data-in' type='checkbox'><label for='data-2d54cdcf-9b44-4720-b151-cdd4cfffd6be' title='Show/Hide data repr'><svg class='icon xr-icon-database'><use xlink:href='#icon-database'></use></svg></label><div class='xr-var-attrs'><dl class='xr-attrs'></dl></div><div class='xr-var-data'><pre>array(146.94301032591366+/-nan, dtype=object)</pre></div></li><li class='xr-var-item'><div class='xr-var-name ], "text/plain": [ "<xarray.Dataset>\n", "Dimensions: ()\n", "Data variables:\n", " BEC_amplitude object 0.0+/-nan\n", " thermal_amplitude object 2104.548431645919+/-nan\n", " BEC_centerx object 146.94301032591366+/-nan\n", " BEC_centery object 147.47224593536436+/-nan\n", " thermal_centerx object 146.27287010988167+/-nan\n", " thermal_centery object 148.78153517037947+/-nan\n", " BEC_sigmax object 17.155488681677085+/-nan\n", " BEC_sigmay object 18.315601451967396+/-nan\n", " thermal_sigmax object 42.999686622150065+/-nan\n", " thermal_sigmay object 51.599623946580074+/-nan\n", " thermalAspectRatio object 1.2+/-nan\n", " condensate_fraction object 0.0+/-nan\n", "Attributes:\n", " IMAGE_SUBCLASS: IMAGE_GRAYSCALE\n", " IMAGE_VERSION: 1.2\n", " IMAGE_WHITE_IS_ZERO: 0\n", " x_start: 810\n", " x_end: 1110\n", " y_end: 1025\n", " y_start: 725\n", " x_center: 960\n", " y_center: 875\n", " x_span: 300\n", " y_span: 300" ] }, "execution_count": 57, "metadata": {}, "output_type": "execute_result" } ], "source": [ "value" ] }, { "cell_type": "code", "execution_count": 17, "metadata": {}, "outputs": [ { "ename": "ValueError", "evalue": "unable to infer dtype on variable 'OD'; xarray cannot serialize arbitrary Python objects", "output_type": "error", "traceback": [ "\u001b[1;31m---------------------------------------------------------------------------\u001b[0m", "\u001b[1;31mValueError\u001b[0m Traceback (most recent call last)", "\u001b[1;32mf:\\Jianshun\\analyseScript\\test.ipynb Cell 25\u001b[0m in \u001b[0;36m1\n\u001b[1;32m----> <a href='vscode-notebook-cell:/f%3A/Jianshun/analyseScript/test.ipynb#Y216sZmlsZQ%3D%3D?line=0'>1</a>\u001b[0m fitResult\u001b[39m.\u001b[39;49mto_netcdf(\u001b[39m\"\u001b[39;49m\u001b[39msaved_on_disk.nc\u001b[39;49m\u001b[39m\"\u001b[39;49m)\n", "File \u001b[1;32m~\\AppData\\Roaming\\Python\\Python39\\site-packages\\xarray\\core\\dataarray.py:3959\u001b[0m, in \u001b[0;36mDataArray.to_netcdf\u001b[1;34m(self, path, mode, format, group, engine, encoding, unlimited_dims, compute, invalid_netcdf)\u001b[0m\n\u001b[0;32m 3955\u001b[0m \u001b[39melse\u001b[39;00m:\n\u001b[0;32m 3956\u001b[0m \u001b[39m# No problems with the name - so we're fine!\u001b[39;00m\n\u001b[0;32m 3957\u001b[0m dataset \u001b[39m=\u001b[39m \u001b[39mself\u001b[39m\u001b[39m.\u001b[39mto_dataset()\n\u001b[1;32m-> 3959\u001b[0m \u001b[39mreturn\u001b[39;00m to_netcdf( \u001b[39m# type: ignore # mypy cannot resolve the overloads:(\u001b[39;49;00m\n\u001b[0;32m 3960\u001b[0m dataset,\n\u001b[0;32m 3961\u001b[0m path,\n\u001b[0;32m 3962\u001b[0m mode\u001b[39m=\u001b[39;49mmode,\n\u001b[0;32m 3963\u001b[0m \u001b[39mformat\u001b[39;49m\u001b[39m=\u001b[39;49m\u001b[39mformat\u001b[39;49m,\n\u001b[0;32m 3964\u001b[0m group\u001b[39m=\u001b[39;49mgroup,\n\u001b[0;32m 3965\u001b[0m engine\u001b[39m=\u001b[39;49mengine,\n\u001b[0;32m 3966\u001b[0m encoding\u001b[39m=\u001b[39;49mencoding,\n\u001b[0;32m 3967\u001b[0m unlimited_dims\u001b[39m=\u001b[39;49munlimited_dims,\n\u001b[0;32m 3968\u001b[0m compute\u001b[39m=\u001b[39;49mcompute,\n\u001b[0;32m 3969\u001b[0m multifile\u001b[39m=\u001b[39;49m\u001b[39mFalse\u001b[39;49;00m,\n\u001b[0;32m 3970\u001b[0m invalid_netcdf\u001b[39m=\u001b[39;49minvalid_netcdf,\n\u001b[0;32m 3971\u001b[0m )\n", "File \u001b[1;32m~\\AppData\\Roaming\\Python\\Python39\\site-packages\\xarray\\backends\\api.py:1216\u001b[0m, in \u001b[0;36mto_netcdf\u001b[1;34m(dataset, path_or_file, mode, format, group, engine, encoding, unlimited_dims, compute, multifile, invalid_netcdf)\u001b[0m\n\u001b[0;32m 1211\u001b[0m \u001b[39m# TODO: figure out how to refactor this logic (here and in save_mfdataset)\u001b[39;00m\n\u001b[0;32m 1212\u001b[0m \u001b[39m# to avoid this mess of conditionals\u001b[39;00m\n\u001b[0;32m 1213\u001b[0m \u001b[39mtry\u001b[39;00m:\n\u001b[0;32m 1214\u001b[0m \u001b[39m# TODO: allow this work (setting up the file for writing array data)\u001b[39;00m\n\u001b[0;32m 1215\u001b[0m \u001b[39m# to be parallelized with dask\u001b[39;00m\n\u001b[1;32m-> 1216\u001b[0m dump_to_store(\n\u001b[0;32m 1217\u001b[0m dataset, store, writer, encoding\u001b[39m=\u001b[39;49mencoding, unlimited_dims\u001b[39m=\u001b[39;49munlimited_dims\n\u001b[0;32m 1218\u001b[0m )\n\u001b[0;32m 1219\u001b[0m \u001b[39mif\u001b[39;00m autoclose:\n\u001b[0;32m 1220\u001b[0m store\u001b[39m.\u001b[39mclose()\n", "File \u001b[1;32m~\\AppData\\Roaming\\Python\\Python39\\site-packages\\xarray\\backends\\api.py:1263\u001b[0m, in \u001b[0;36mdump_to_store\u001b[1;34m(dataset, store, writer, encoder, encoding, unlimited_dims)\u001b[0m\n\u001b[0;32m 1260\u001b[0m \u001b[39mif\u001b[39;00m encoder:\n\u001b[0;32m 1261\u001b[0m variables, attrs \u001b[39m=\u001b[39m encoder(variables, attrs)\n\u001b[1;32m-> 1263\u001b[0m store\u001b[39m.\u001b[39;49mstore(variables, attrs, check_encoding, writer, unlimited_dims\u001b[39m=\u001b[39;49munlimited_dims)\n", "File \u001b[1;32m~\\AppData\\Roaming\\Python\\Python39\\site-packages\\xarray\\backends\\common.py:269\u001b[0m, in \u001b[0;36mAbstractWritableDataStore.store\u001b[1;34m(self, variables, attributes, check_encoding_set, writer, unlimited_dims)\u001b[0m\n\u001b[0;32m 266\u001b[0m \u001b[39mif\u001b[39;00m writer \u001b[39mis\u001b[39;00m \u001b[39mNone\u001b[39;00m:\n\u001b[0;32m 267\u001b[0m writer \u001b[39m=\u001b[39m ArrayWriter()\n\u001b[1;32m--> 269\u001b[0m variables, attributes \u001b[39m=\u001b[39m \u001b[39mself\u001b[39;49m\u001b[39m.\u001b[39;49mencode(variables, attributes)\n\u001b[0;32m 271\u001b[0m \u001b[39mself\u001b[39m\u001b[39m.\u001b[39mset_attributes(attributes)\n\u001b[0;32m 272\u001b[0m \u001b[39mself\u001b[39m\u001b[39m.\u001b[39mset_dimensions(variables, unlimited_dims\u001b[39m=\u001b[39munlimited_dims)\n", "File \u001b[1;32m~\\AppData\\Roaming\\Python\\Python39\\site-packages\\xarray\\backends\\common.py:358\u001b[0m, in \u001b[0;36mWritableCFDataStore.encode\u001b[1;34m(self, variables, attributes)\u001b[0m\n\u001b[0;32m 355\u001b[0m \u001b[39mdef\u001b[39;00m \u001b[39mencode\u001b[39m(\u001b[39mself\u001b[39m, variables, attributes):\n\u001b[0;32m 356\u001b[0m \u001b[39m# All NetCDF files get CF encoded by default, without this attempting\u001b[39;00m\n\u001b[0;32m 357\u001b[0m \u001b[39m# to write times, for example, would fail.\u001b[39;00m\n\u001b[1;32m--> 358\u001b[0m variables, attributes \u001b[39m=\u001b[39m cf_encoder(variables, attributes)\n\u001b[0;32m 359\u001b[0m variables \u001b[39m=\u001b[39m {k: \u001b[39mself\u001b[39m\u001b[39m.\u001b[39mencode_variable(v) \u001b[39mfor\u001b[39;00m k, v \u001b[39min\u001b[39;00m variables\u001b[39m.\u001b[39mitems()}\n\u001b[0;32m 360\u001b[0m attributes \u001b[39m=\u001b[39m {k: \u001b[39mself\u001b[39m\u001b[39m.\u001b[39mencode_attribute(v) \u001b[39mfor\u001b[39;00m k, v \u001b[39min\u001b[39;00m attributes\u001b[39m.\u001b[39mitems()}\n", "File \u001b[1;32m~\\AppData\\Roaming\\Python\\Python39\\site-packages\\xarray\\conventions.py:775\u001b[0m, in \u001b[0;36mcf_encoder\u001b[1;34m(variables, attributes)\u001b[0m\n\u001b[0;32m 772\u001b[0m \u001b[39m# add encoding for time bounds variables if present.\u001b[39;00m\n\u001b[0;32m 773\u001b[0m _update_bounds_encoding(variables)\n\u001b[1;32m--> 775\u001b[0m new_vars \u001b[39m=\u001b[39m {k: encode_cf_variable(v, name\u001b[39m=\u001b[39mk) \u001b[39mfor\u001b[39;00m k, v \u001b[39min\u001b[39;00m variables\u001b[39m.\u001b[39mitems()}\n\u001b[0;32m 777\u001b[0m \u001b[39m# Remove attrs from bounds variables (issue #2921)\u001b[39;00m\n\u001b[0;32m 778\u001b[0m \u001b[39mfor\u001b[39;00m var \u001b[39min\u001b[39;00m new_vars\u001b[39m.\u001b[39mvalues():\n", "File \u001b[1;32m~\\AppData\\Roaming\\Python\\Python39\\site-packages\\xarray\\conventions.py:775\u001b[0m, in \u001b[0;36m<dictcomp>\u001b[1;34m(.0)\u001b[0m\n\u001b[0;32m 772\u001b[0m \u001b[39m# add encoding for time bounds variables if present.\u001b[39;00m\n\u001b[0;32m 773\u001b[0m _update_bounds_encoding(variables)\n\u001b[1;32m--> 775\u001b[0m new_vars \u001b[39m=\u001b[39m {k: encode_cf_variable(v, name\u001b[39m=\u001b[39;49mk) \u001b[39mfor\u001b[39;00m k, v \u001b[39min\u001b[39;00m variables\u001b[39m.\u001b[39mitems()}\n\u001b[0;32m 777\u001b[0m \u001b[39m# Remove attrs from bounds variables (issue #2921)\u001b[39;00m\n\u001b[0;32m 778\u001b[0m \u001b[39mfor\u001b[39;00m var \u001b[39min\u001b[39;00m new_vars\u001b[39m.\u001b[39mvalues():\n", "File \u001b[1;32m~\\AppData\\Roaming\\Python\\Python39\\site-packages\\xarray\\conventions.py:189\u001b[0m, in \u001b[0;36mencode_cf_variable\u001b[1;34m(var, needs_copy, name)\u001b[0m\n\u001b[0;32m 186\u001b[0m var \u001b[39m=\u001b[39m coder\u001b[39m.\u001b[39mencode(var, name\u001b[39m=\u001b[39mname)\n\u001b[0;32m 188\u001b[0m \u001b[39m# TODO(kmuehlbauer): check if ensure_dtype_not_object can be moved to backends:\u001b[39;00m\n\u001b[1;32m--> 189\u001b[0m var \u001b[39m=\u001b[39m ensure_dtype_not_object(var, name\u001b[39m=\u001b[39;49mname)\n\u001b[0;32m 191\u001b[0m \u001b[39mfor\u001b[39;00m attr_name \u001b[39min\u001b[39;00m CF_RELATED_DATA:\n\u001b[0;32m 192\u001b[0m pop_to(var\u001b[39m.\u001b[39mencoding, var\u001b[39m.\u001b[39mattrs, attr_name)\n", "File \u001b[1;32m~\\AppData\\Roaming\\Python\\Python39\\site-packages\\xarray\\conventions.py:145\u001b[0m, in \u001b[0;36mensure_dtype_not_object\u001b[1;34m(var, name)\u001b[0m\n\u001b[0;32m 143\u001b[0m data[missing] \u001b[39m=\u001b[39m fill_value\n\u001b[0;32m 144\u001b[0m \u001b[39melse\u001b[39;00m:\n\u001b[1;32m--> 145\u001b[0m data \u001b[39m=\u001b[39m _copy_with_dtype(data, dtype\u001b[39m=\u001b[39m_infer_dtype(data, name))\n\u001b[0;32m 147\u001b[0m \u001b[39massert\u001b[39;00m data\u001b[39m.\u001b[39mdtype\u001b[39m.\u001b[39mkind \u001b[39m!=\u001b[39m \u001b[39m\"\u001b[39m\u001b[39mO\u001b[39m\u001b[39m\"\u001b[39m \u001b[39mor\u001b[39;00m data\u001b[39m.\u001b[39mdtype\u001b[39m.\u001b[39mmetadata\n\u001b[0;32m 148\u001b[0m var \u001b[39m=\u001b[39m Variable(dims, data, attrs, encoding, fastpath\u001b[39m=\u001b[39m\u001b[39mTrue\u001b[39;00m)\n", "File \u001b[1;32m~\\AppData\\Roaming\\Python\\Python39\\site-packages\\xarray\\conventions.py:77\u001b[0m, in \u001b[0;36m_infer_dtype\u001b[1;34m(array, name)\u001b[0m\n\u001b[0;32m 74\u001b[0m \u001b[39mif\u001b[39;00m dtype\u001b[39m.\u001b[39mkind \u001b[39m!=\u001b[39m \u001b[39m\"\u001b[39m\u001b[39mO\u001b[39m\u001b[39m\"\u001b[39m:\n\u001b[0;32m 75\u001b[0m \u001b[39mreturn\u001b[39;00m dtype\n\u001b[1;32m---> 77\u001b[0m \u001b[39mraise\u001b[39;00m \u001b[39mValueError\u001b[39;00m(\n\u001b[0;32m 78\u001b[0m \u001b[39m\"\u001b[39m\u001b[39munable to infer dtype on variable \u001b[39m\u001b[39m{!r}\u001b[39;00m\u001b[39m; xarray \u001b[39m\u001b[39m\"\u001b[39m\n\u001b[0;32m 79\u001b[0m \u001b[39m\"\u001b[39m\u001b[39mcannot serialize arbitrary Python objects\u001b[39m\u001b[39m\"\u001b[39m\u001b[39m.\u001b[39mformat(name)\n\u001b[0;32m 80\u001b[0m )\n", "\u001b[1;31mValueError\u001b[0m: unable to infer dtype on variable 'OD'; xarray cannot serialize arbitrary Python objects" ] } ], "source": [ "fitResult.to_netcdf(\"saved_on_disk.nc\")" ] }, { "attachments": {}, "cell_type": "markdown", "metadata": {}, "source": [ "# Get the Ncount" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "Ncount = dataSet_crop.OD.sum(dim=(scanAxis[0], 'x', 'y'))" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "Ncount.load()\n", "\n", "fig = plt.figure()\n", "ax = fig.gca()\n", "Ncount.plot(ax=ax)" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "fitAnalyser = FitAnalyser(\"Lorentzian With Offset\")\n", "params = fitAnalyser.guess(Ncount, x='runs', dask=\"parallelized\", guess_kwargs=dict(negative=True))" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "fitResult = fitAnalyser.fit(Ncount, params, x='runs', dask=\"parallelized\")\n", "fitCurve = fitAnalyser.eval(fitResult, x=np.arange(40), dask=\"parallelized\").load()" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "fig = plt.figure()\n", "ax = fig.gca()\n", "plt.errorbar([1], [1], yerr=[1])" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "fitCurve.plot.errorbar(yerr=fitCurve)" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "np.ufunc(fitCurve)" ] }, { "attachments": {}, "cell_type": "markdown", "metadata": {}, "source": [ "# Read CSV" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "# filePath = 'Z:/Dy_Lab/Data/Measurements/Experiments/DyBEC/BEC Stability Check/20230509-0007/*.csv'\n", "\n", "# filePath = np.sort(glob.glob(filePath))\n", "\n", "# read_csv_file(filePath, maxFileNum=5, csvEngine='pandas', csvKwargs=dict(header=[0,1], na_filter=False, index_col=0))\n", "# read_csv_file(filePath, csvEngine='dask')" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "filePath = 'Z:/Dy_Lab/Data/Measurements/Experiments/DyBEC/BEC Stability Check/20230509-0007/*.csv'\n", "\n", "filePath = np.sort(glob.glob(filePath))\n", "\n", "data = np.empty(filePath.shape,dtype=object)\n", "\n", "i = 0\n", "for fp in filePath:\n", " data_single = pd.read_csv(fp)\n", " data_single = xr.Dataset.from_dataframe(data_single)\n", " data_single = data_single.drop_isel(index=0)\n", " # data_single = data_single.expand_dims(dim='runs')\n", " data[i] = data_single\n", " i = i + 1\n", "\n", "data = xr.concat(data, 'runs')\n", "\n", "data = data.assign_coords(dict(index=data.Time.isel(runs=0).astype(float))).rename(dict(index='time')).astype(float)" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "data" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "arm2_mean = data['Channel A'].mean(dim='runs')\n", "arm2_std = data['Channel A'].std(dim='runs')" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "arm2_mean.plot.errorbar(yerr=arm2_std, fmt='ob')" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "arm2_std.plot.errorbar(fmt='ob')" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "data['Channel A'].sel(time=4.55, method='nearest').plot.errorbar(fmt='ob')\n", "\n", "plt.ylim([0, 0.15])\n", "plt.show()" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [] } ], "metadata": { "kernelspec": { "display_name": "env", "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.9.12" }, "orig_nbformat": 4, "vscode": { "interpreter": { "hash": "c05913ad4f24fdc6b2418069394dc5835b1981849b107c9ba6df693aafd66650" } } }, "nbformat": 4, "nbformat_minor": 2 }
|