diff options
| author |  Weiqun Zhang <weiqunzhang@lbl.gov>
| 2017-06-01 18:02:25 -0700 |
|---|---|---|
| committer | Weiqun Zhang <weiqunzhang@lbl.gov>
| 2017-06-01 18:02:25 -0700 |
| commit | 204551745ca65d27e140d83d0c2e79dfdb2d9225 (patch) | |
| tree | 9d79d77cdb9500cafa0f56b434b4d7fe1f00d476 /Example/Larmor/plot_particle_path.py | |
| parent | 9f05ae6cfd8153c705df186758cb13423fbd493a (diff) | |
| download | WarpX-204551745ca65d27e140d83d0c2e79dfdb2d9225.tar.gz WarpX-204551745ca65d27e140d83d0c2e79dfdb2d9225.tar.zst WarpX-204551745ca65d27e140d83d0c2e79dfdb2d9225.zip | |
update python script
Diffstat (limited to 'Example/Larmor/plot_particle_path.py')
| -rw-r--r-- | Example/Larmor/plot_particle_path.py | 20 |
1 files changed, 13 insertions, 7 deletions
diff --git a/Example/Larmor/plot_particle_path.py b/Example/Larmor/plot_particle_path.py index c925e1728..bc0490859 100644 --- a/Example/Larmor/plot_particle_path.py +++ b/Example/Larmor/plot_particle_path.py @@ -116,21 +116,27 @@ if __name__ == "__main__": import pylab as plt import glob - x = [] - y = [] + x0 = [] + y0 = [] + x1 = [] + y1 = [] fn_list = glob.glob("plt?????") fn_list.sort() for fn in fn_list: - idata, rdata = read_particle_data(fn) - x.append(rdata[0][0]) - y.append(rdata[0][1]) + idata, rdata = read_particle_data(fn, ptype="particle0") + x0.append(rdata[0][0]) + y0.append(rdata[0][1]) + idata, rdata = read_particle_data(fn, ptype="particle1") + x1.append(rdata[0][0]) + y1.append(rdata[0][1]) fig = plt.gcf() fig.set_size_inches(8, 8) - plt.plot(x, y, '.') - plt.axis((-30e-6, 30e-6, -30e-6, 30e-6)) + plt.plot(x0, y0, 'r.') + plt.plot(x1, y1, 'b.') + plt.axis((-2., 2., -2., 2.)) ax = plt.gca() ax.set_xlabel(r'$x$') ax.set_ylabel(r'$y$') |