diff options
| author |  Yinjian Zhao <yinjianzhao@lbl.gov>
| 2020-09-29 22:19:01 -0700 |
|---|---|---|
| committer |  GitHub <noreply@github.com>
| 2020-09-29 22:19:01 -0700 |
| commit | d3872b3f9a313cff14a44dc14964aec4bc26c47d (patch) | |
| tree | d039f7e3aff2da603870ab41b6741943d4b3b84b /Tools/PostProcessing/Visualization.ipynb | |
| parent | 81195bd82440fe206552f0306767283606d44330 (diff) | |
| download | WarpX-d3872b3f9a313cff14a44dc14964aec4bc26c47d.tar.gz WarpX-d3872b3f9a313cff14a44dc14964aec4bc26c47d.tar.zst WarpX-d3872b3f9a313cff14a44dc14964aec4bc26c47d.zip | |
Minor modification of Visualization.ipynb (#1383)
* Update inputs_3d
* Apply suggestions from code review
Co-authored-by: Axel Huebl <axel.huebl@plasma.ninja>
Diffstat (limited to '')
| -rw-r--r-- | Tools/PostProcessing/Visualization.ipynb | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/Tools/PostProcessing/Visualization.ipynb b/Tools/PostProcessing/Visualization.ipynb index 8dd2f3441..dbeb9a734 100644 --- a/Tools/PostProcessing/Visualization.ipynb +++ b/Tools/PostProcessing/Visualization.ipynb @@ -45,7 +45,9 @@ "metadata": {}, "outputs": [], "source": [ - "plotfile = './diags/plotfiles/plt00001'\n", + "diag_name = 'diag' # E.g., diagnostics.diags_names = diag\n", + "iteration = 0\n", + "plotfile = './diags/{}{:05d}'.format(diag_name, iteration)\n", "field = 'Ex'\n", "species = 'electron'\n", "ds = yt.load( plotfile ) # Load the plotfile\n", @@ -94,11 +96,11 @@ "# Get particle quantities\n", "ad = ds.all_data()\n", "x = ad[species, 'particle_position_x'].v\n", - "z = ad[species, 'particle_position_y'].v\n", + "z = ad[species, 'particle_position_z'].v\n", "\n", "# Plot image\n", "plt.figure()\n", - "plt.imshow(Bx, extent=extent)\n", + "plt.imshow(Bx[:,Bx.shape[1]//2,:], extent=extent, aspect='auto')\n", "plt.scatter(z,x,s=.1,c='k')" ] }, @@ -132,7 +134,7 @@ "header = './lab_frame_data/Header'\n", "allrd, info = read_raw_data.read_lab_snapshot(snapshot, header) # Read field data\n", "F = allrd[field]\n", - "print( \"Available info: \", *list(info.keys()) )\n", + "print( \"Available info: \", list(info.keys()) )\n", "print(\"Available fields: \", info['field_names'])\n", "nx = info['nx']\n", "nz = info['nz']\n", @@ -145,7 +147,7 @@ "\n", "plt.figure(figsize=(6, 3))\n", "extent = np.array([info['zmin'], info['zmax'], info['xmin'], info['xmax']])\n", - "plt.imshow(F, aspect='auto', extent=extent, cmap='seismic')\n", + "plt.imshow(F[:,F.shape[1]//2,:], aspect='auto', extent=extent, cmap='seismic')\n", "plt.colorbar()\n", "plt.plot(zbo, xbo, 'g.', markersize=1.)\n" ] |