gw and plot

This commit is contained in:
2024-11-12 17:11:32 +08:00
parent 1d8159f481
commit 287b13c529
3 changed files with 70 additions and 4 deletions

View File

@ -6,8 +6,59 @@
"metadata": {},
"outputs": [],
"source": [
"from awkward import Record as AwkwardRecord, Array as AwkwardArray\n",
"import awkward as ak"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"FILENAME = \"export-2024-11-12-16-59-20.parquet\"\n",
"data = ak.from_parquet(FILENAME)[0]\n",
"display(data)"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"from typing import Literal\n",
"import plotly.graph_objects as go\n",
"import plotly.express as px\n",
"import numpy as np\n",
"\n",
"\n",
"def to_scatter(key: str, dev_history: AwkwardArray, plot_key: Literal[\"value\", \"rssi\"] = \"value\") -> go.Scatter:\n",
" x = ak.to_numpy(dev_history[\"time\"])\n",
" if plot_key == \"rssi\":\n",
" y = ak.to_numpy(dev_history[\"rssi\"])\n",
" elif plot_key == \"value\":\n",
" y = ak.to_numpy(dev_history[\"value\"])\n",
" else:\n",
" raise ValueError(f\"Unknown plot_key: {plot_key}\")\n",
" return go.Scatter(x=x, y=y, mode=\"lines+markers\", name=key)\n",
"\n",
"scatters: list[go.Scatter] = []\n",
"for k in data.fields:\n",
" val = data[k]\n",
" scatters.append(to_scatter(k, val, \"value\"))\n",
"\n",
"fig = go.Figure(scatters)"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"display(fig)"
]
}
],
"metadata": {
@ -17,7 +68,15 @@
"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.12.7"
}
},