feat(viz): Update plotting for STFT data visualization with color map 'jet' and added color bar

This commit is contained in:
nuluh
2025-05-29 20:35:35 +07:00
parent d151062115
commit 254b24cb21

View File

@@ -334,8 +334,9 @@
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
"source": [ "source": [
"len(ready_data1a)\n", "# len(ready_data1a)\n",
"# plt.pcolormesh(ready_data1[0])" "# plt.pcolormesh(ready_data1[0])\n",
"ready_data1a[0].max().max()"
] ]
}, },
{ {
@@ -345,7 +346,8 @@
"outputs": [], "outputs": [],
"source": [ "source": [
"for i in range(6):\n", "for i in range(6):\n",
" plt.pcolormesh(ready_data1a[i])\n", " plt.pcolormesh(ready_data1a[i], cmap=\"jet\", vmax=0.03, vmin=0.0)\n",
" plt.colorbar() \n",
" plt.title(f'STFT Magnitude for case {i} sensor 1')\n", " plt.title(f'STFT Magnitude for case {i} sensor 1')\n",
" plt.xlabel(f'Frequency [Hz]')\n", " plt.xlabel(f'Frequency [Hz]')\n",
" plt.ylabel(f'Time [sec]')\n", " plt.ylabel(f'Time [sec]')\n",