26 lines
928 B
Python
26 lines
928 B
Python
|
__author__ = 'dmy'
|
||
|
|
|
||
|
|
import numpy
|
||
|
|
|
||
|
|
import scipy.sparse as sparse
|
||
|
|
|
||
|
|
import scipy.sparse
|
||
|
|
|
||
|
|
def FormYMatrix(lineBlock,transBlock,busNum):
|
||
|
|
r=lineBlock[:,3]
|
||
|
|
x=lineBlock[:,4]
|
||
|
|
gb=1/(r+1j*x)
|
||
|
|
y=sparse.coo_matrix((gb,(lineBlock[:,1]-1,lineBlock[:,2]-1)),shape=(busNum,busNum))
|
||
|
|
#加接地支路
|
||
|
|
b2=lineBlock[:,5]
|
||
|
|
y+=sparse.coo_matrix((1j*b2,(lineBlock[:,1]-1,lineBlock[:,1]-1)),shape=(busNum,busNum))
|
||
|
|
y+=sparse.coo_matrix((1j*b2,(lineBlock[:,2]-1,lineBlock[:,2]-1)),shape=(busNum,busNum))
|
||
|
|
print(y.sum(1))
|
||
|
|
print(numpy.arange(int(busNum)).shape[0])
|
||
|
|
y+=sparse.coo_matrix((numpy.array(y.sum(1)),(numpy.arange(int(busNum)),numpy.arange(int(busNum)))),shape=(busNum,busNum))
|
||
|
|
#加变压器
|
||
|
|
#TODO:没有考虑变比
|
||
|
|
transX=transBlock[:,3]
|
||
|
|
transR=transBlock[:,4]
|
||
|
|
transGB=1/(transR+1j*transX)
|
||
|
|
y+=sparse.coo_matrix((transGB,(transBlock[:,1]-1,transBlock[:,2]-1)),shape=(busNum,busNum))
|