修复几个边界条件bug

main.py

@@ -1,3 +1,5 @@
+import numpy as np
+
 from core import *
 import timeit
 
@@ -5,13 +7,13 @@ import timeit
 def egm():
     h_g_avr_sag = 11.67 * 2 / 3
     h_c_avr_sag = 14.43 * 2 / 3
-    h_whole = 140  # 杆塔全高
+    h_whole = 260  # 杆塔全高
     voltage_n = 3  # 工作电压分成多少份来计算
     td = 20  # 雷暴日
     insulator_c_len = 6.8  # 串子绝缘长度
     string_c_len = 9.2
     string_g_len = 0.5
-    gc_x = [17.9, 16, 15, 16]
+    gc_x = [17.9, 17, 15, 17]
 
     # 以后考虑地形角度,地面线
     def ground_surface(x):
@@ -34,6 +36,7 @@ def egm():
     rg_x = None
     rg_y = None
     cad = Draw()
+    n_sf_phases = np.zeros((phase_n, voltage_n))  # 计算每一相的跳闸率
     for phase_conductor in range(phase_n):
         rs_x = gc_x[phase_conductor]
         rs_y = gc_y[phase_conductor]
@@ -149,7 +152,7 @@ def egm():
                     if distance > rc:
                         print("暴露弧已经完全被屏蔽")
                         break
-            if phase_conductor == 1:
+            if phase_conductor == 2:
                 cad.draw(i_min, u_ph, rs_x, rs_y, rc_x, rc_y, rg_x, rg_y, rg_type, 2)
                 cad.draw(i_max, u_ph, rs_x, rs_y, rc_x, rc_y, rg_x, rg_y, rg_type, 6)
                 cad.save()
@@ -205,8 +208,10 @@ def egm():
                 2 * ng / 10 * calculus
             )  # 跳闸率 利用Q╱GDW 11452-2015 架空输电线路防雷导则的公式 Ng=0.023*Td^(1.3) 20天雷暴日地闪密度为1.13
             avr_n_sf += n_sf / voltage_n
+            n_sf_phases[phase_conductor][u_bar] = n_sf
             print(f"工作电压为{u_ph:.2f}kV时,跳闸率是{n_sf:.6}")
     print(f"跳闸率是{avr_n_sf:.6f}")
+    print(f"不同相跳闸率是{np.mean(n_sf_phases,axis=1)}")
 
 
 def speed():