{ "cells": [ { "cell_type": "code", "execution_count": 2, "metadata": { "collapsed": true }, "outputs": [], "source": [ "import pandas as pd\n", "import numpy as np\n", "import matplotlib.pyplot as plt\n", "import scipy\n", "from glob import glob" ] }, { "cell_type": "code", "execution_count": 10, "outputs": [], "source": [ "def read_xy(path: str):\n", " df = pd.read_csv(path)\n", " df['N'] = df.index + 1\n", " df['r'] = (df.x ** 2 + df.y ** 2) ** 0.5\n", " df['cr'] = df.r.cummax()\n", " df['fd'] = np.log(df.N) / np.log(df.cr)\n", "\n", " return df\n", "\n", "\n", "def read_r3(path: str):\n", " df = pd.read_csv(path)\n", " df['N'] = df.index + 1\n", " df['r'] = (df.r0 ** 2 + df.r1 ** 2 + df.r2 ** 2) ** 0.5\n", " df['cr'] = df.r.cummax()\n", " df['fd'] = np.log(df.N) / np.log(df.cr)\n", "\n", " return df" ], "metadata": { "collapsed": false } }, { "cell_type": "code", "execution_count": 13, "outputs": [], "source": [ "def read_all(load_dir: str, reader):\n", " paths = glob(f'{load_dir}/*.csv')\n", " return [reader(path) for path in paths]" ], "metadata": { "collapsed": false } }, { "cell_type": "code", "execution_count": 5, "outputs": [ { "name": "stderr", "output_type": "stream", "text": [ "/Users/joshuacoles/Library/Caches/pypoetry/virtualenvs/data-analysis-B4Au_hWl-py3.10/lib/python3.10/site-packages/pandas/core/arraylike.py:402: RuntimeWarning: divide by zero encountered in log\n", " result = getattr(ufunc, method)(*inputs, **kwargs)\n" ] }, { "data": { "text/plain": " r0 r1 r2 N r cr fd\n0 0 0 0 1 0.000000 0.000000 -0.000000\n1 1 0 0 2 1.000000 1.000000 inf\n2 0 0 1 3 1.000000 1.000000 inf\n3 -1 0 1 4 1.414214 1.414214 4.000000\n4 1 0 1 5 1.414214 1.414214 4.643856\n... .. .. .. ... ... ... ...\n1995 66 13 30 1996 73.654599 76.720271 1.750832\n1996 56 19 29 1997 65.863495 76.720271 1.750947\n1997 68 17 26 1998 74.759615 76.720271 1.751063\n1998 52 22 36 1999 66.962676 76.720271 1.751178\n1999 62 6 17 2000 64.567794 76.720271 1.751293\n\n[2000 rows x 7 columns]", "text/html": "
| \n | r0 | \nr1 | \nr2 | \nN | \nr | \ncr | \nfd | \n
|---|---|---|---|---|---|---|---|
| 0 | \n0 | \n0 | \n0 | \n1 | \n0.000000 | \n0.000000 | \n-0.000000 | \n
| 1 | \n1 | \n0 | \n0 | \n2 | \n1.000000 | \n1.000000 | \ninf | \n
| 2 | \n0 | \n0 | \n1 | \n3 | \n1.000000 | \n1.000000 | \ninf | \n
| 3 | \n-1 | \n0 | \n1 | \n4 | \n1.414214 | \n1.414214 | \n4.000000 | \n
| 4 | \n1 | \n0 | \n1 | \n5 | \n1.414214 | \n1.414214 | \n4.643856 | \n
| ... | \n... | \n... | \n... | \n... | \n... | \n... | \n... | \n
| 1995 | \n66 | \n13 | \n30 | \n1996 | \n73.654599 | \n76.720271 | \n1.750832 | \n
| 1996 | \n56 | \n19 | \n29 | \n1997 | \n65.863495 | \n76.720271 | \n1.750947 | \n
| 1997 | \n68 | \n17 | \n26 | \n1998 | \n74.759615 | \n76.720271 | \n1.751063 | \n
| 1998 | \n52 | \n22 | \n36 | \n1999 | \n66.962676 | \n76.720271 | \n1.751178 | \n
| 1999 | \n62 | \n6 | \n17 | \n2000 | \n64.567794 | \n76.720271 | \n1.751293 | \n
2000 rows × 7 columns
\n| \n | r0 | \nr1 | \nr2 | \nN | \nr | \ncr | \nfd | \n
|---|---|---|---|---|---|---|---|
| 3832 | \n0 | \n0 | \n10 | \n3833 | \n10.0 | \n60.241182 | \n2.013345 | \n
| 3854 | \n0 | \n0 | \n0 | \n3855 | \n0.0 | \n60.241182 | \n2.014741 | \n
| 3865 | \n0 | \n0 | \n2 | \n3866 | \n2.0 | \n60.241182 | \n2.015436 | \n
| 5818 | \n0 | \n0 | \n-12 | \n5819 | \n12.0 | \n60.241182 | \n2.115210 | \n