change: gcr fig

README.md

@@ -57,9 +57,7 @@ Dose Estimation by Simulation of China Space Station
 
 考虑异常部分，我们使用`SPENVIS`+`CREME86`模型进行模拟，分别获取银河宇宙射线通量的最佳近似值（`M=1`）以及`90%`最坏情况下的银河宇宙射线通量（即实际情况大约只有`10%`的可能通量大于这一估计值，`M=3`）。
 
-<div align=center>
-<img src="docs/spectra-M1.png" style="max-width: 45%;margin-right: 1%;"><img src="docs/spectra-M3.png" style="max-width: 45%;margin-left: 1%;">
-</div>
+<div align=center><img src="docs/spectra.png" style="max-width: 80%;"></div>
 
 #### 太阳辐射[^3]
 太阳辐射的主要来源是日冕抛射和太阳耀斑，具有明显的周期性，主要的周期有11年、22年等。

utils/draw.py

@@ -42,15 +42,20 @@ def orbit():
     plt.savefig('docs/orbit.png', bbox_inches='tight')
 
 
-def GCR(M=1):
+def GCR(M=1, ax=None):
     _, data = data_read('assets/CREME86-M{:d}.csv'.format(M))
     E = data['Energy']
     proton_i, proton_d = data['IFlux_1'], data['DFlux_1']
     alpha_i, alpha_d = data['IFlux_2'], data['DFlux_2']
 
-    fig, ax1 = plt.subplots(1, 1)
+    if ax:
+        ax1 = ax
+    else:
+        _, ax1 = plt.subplots(1, 1)
+
     ax1.plot(E, proton_i, label=r'$p$')
     ax1.plot(E, alpha_i, label=r'$\alpha$')
+    ax1.set_ylim([1, 5 * 1e2])
     ax1.set_yscale('log')
     ax1.set_ylabel(r'$Integrated\ Flux\ (\mathrm{m^{-2}sr^{-1}s^{-1}})$')
     ax1.legend(loc="upper left")
@@ -58,15 +63,28 @@ def GCR(M=1):
     ax2 = ax1.twinx()
     ax2.plot(E, proton_d, linestyle=':', label=r'$p$')
     ax2.plot(E, alpha_d, linestyle=':', label=r'$\alpha$')
+    ax2.set_ylim([1e-5, 5 * 1e-2])
     ax2.set_yscale('log')
     ax2.set_ylabel(r'$Differential\ Flux\ (\mathrm{m^{-2}sr^{-1}s^{-1}MeV^{-1}})$')
     ax2.legend(loc="upper right")
 
     ax1.set_xlabel(r'$Energy\ (\mathrm{MeV})$')
     ax1.set_xlim(1, 1e5)
-    plt.title('Average spectra - M = {:d}'.format(M))
-    plt.xscale('log')
-    plt.savefig('docs/spectra-M{:d}'.format(M))
+    
+    if ax is None:
+        plt.title('Average spectra - M = {:d}'.format(M))
+        plt.xscale('log')
+        plt.savefig('docs/spectra-M{:d}.png'.format(M), bbox_inches='tight')
+    else:
+        ax.set_title('Average spectra - M = {:d}'.format(M))
+        ax.set_xscale('log')
 
-GCR()
+fig = plt.figure(figsize=(20, 10), dpi=150)
+
+GCR(1, fig.add_subplot(1, 2, 1))
+GCR(3, fig.add_subplot(1, 2, 2))
+
+fig.savefig('docs/spectra.png', bbox_inches='tight')
+
+GCR(1)
 GCR(3)