From 8cbcc431f17af9eb094d4f7690e553846d41484c Mon Sep 17 00:00:00 2001 From: YiHui Liu Date: Wed, 27 Jul 2022 09:36:15 +0800 Subject: [PATCH] add: axis label --- main.py | 21 +++++++++++++++++++++ qdx/__init__.py | 2 ++ qdx/process.py | 8 ++++++++ 3 files changed, 31 insertions(+) create mode 100644 main.py diff --git a/main.py b/main.py new file mode 100644 index 0000000..051a131 --- /dev/null +++ b/main.py @@ -0,0 +1,21 @@ +import numpy as np +from qdx import Process + +p1 = Process("coef1.csv", n=5) +p2 = Process("coef2.csv", n=5) + +# Mg25 + d -> Mg26* + p +angles = [i for i in np.arange(8, 64, 4)] + [10] +for angle in angles: + task = "Mg25(dp)Mg26/angle-{:d}.csv".format(angle) + print("Processing Task :", task) + + p1.read_data(task) + p1.energy_filter(12, 30) + p1.draw_filter("Mg25(dp)Mg26/figure/filter/angle-{:d}.png".format(angle)) + p1.draw_result("Mg25(dp)Mg26/figure/result/angle-{:d}.png".format(angle)) + + p2.read_data(task) + p2.energy_filter(12, 30) + p2.draw_filter("Mg25(dp)Mg26/figure/filter/angle-{:d}-o.png") + p2.draw_result("Mg25(dp)Mg26/figure/result/angle-{:d}-o.png".format(angle)) diff --git a/qdx/__init__.py b/qdx/__init__.py index 9497f96..12d66f4 100644 --- a/qdx/__init__.py +++ b/qdx/__init__.py @@ -1 +1,3 @@ from .Bind import Bind +from .process import Process +from .calibration import Calibration diff --git a/qdx/process.py b/qdx/process.py index 9382e55..7ebc0d7 100644 --- a/qdx/process.py +++ b/qdx/process.py @@ -123,6 +123,9 @@ class Process(object): ax.scatter(self.pX_n, self.eng_n, s=0.01, color="orange") ax.plot(px_x, self.reg(px_x)) + ax.set_xlabel("x (mm)") + ax.set_ylabel("Energy (MeV)") + fig.savefig(path) plt.close() @@ -147,5 +150,10 @@ class Process(object): ax1.set_xticks(np.arange(px_min, px_max, 50)) ax2.set_xticks(np.arange(px_min, px_max, 50)) + ax1.set_xlabel("x (mm)") + ax1.set_ylabel("Energy (MeV)") + ax2.set_xlabel("x (mm)") + ax2.set_ylabel("Count per bin") + fig.savefig(path) plt.close()