diff --git a/OPF.m b/OPF.m
index 5828b3d..165d614 100644
--- a/OPF.m
+++ b/OPF.m
@@ -88,7 +88,10 @@ absPDLoad=abs(  (PD(Loadi)-PDReal(Loadi))./PDReal(Loadi)  );
 maxPDError=max(absPDLoad);
 disp('index');
 Loadi(absPDLoad==maxPDError);
-%% 计算线损
+%% 计算总线损
+totalLoss=(PGBal(Balance)-sum(PD(Loadi)))*100;
+fprintf('总的损耗为%f(MW 有名值)\n',totalLoss);
+%% 计算各线损
 Lineloss(Linei,Linej,Liner,Linex,Lineb,Transfori,Transforj,Transforr,Transforx,Transfork0,Volt,UAngel);
 toc  
 
diff --git a/OPF_Init.m b/OPF_Init.m
index aca5a39..d24a221 100644
--- a/OPF_Init.m
+++ b/OPF_Init.m
@@ -1,5 +1,5 @@
 function [Volt,UAngel,Init_Z,Init_W,Init_L,Init_U,Init_Y,PG,QG,RestraintCount,wG,wD,PD,PD0,randPDind,Loadi]=OPF_Init(Busnum,Balance,PG,QG,Volt,GenU,GenL,PVi,PGi,PVQU,PVQL,PD0,QD,PD)
-Loadi=find(QD~=0 & PD~=0);
+Loadi=find(QD~=0 | PD~=0);
 %Loadi=[1:Busnum]';
 RestraintCount=size(PVi,1)+size(PGi,1)+size(Loadi,1)+Busnum*1; %约束条件数
 t_Bal_volt=Volt(Balance);