FDL.py 文件源码

def XB_ground(Y,index_PQ, index_P, n_PQ, n_P):
    case_number, _ = np.shape(Y)
    _, B = util.get_G_B(Y)
    X_p = np.zeros((case_number, case_number))
    B_p = np.zeros((n_P, n_P))
    B_pp = np.zeros((n_PQ, n_PQ))
    g_b_round = np.zeros(case_number)
    for i in xrange(case_number):
        a = np.sum(Y[i])
        if LA.norm(a) > 1e-5:
            g_b_round[i] = np.reciprocal(np.imag(np.reciprocal(a)))

    for i in xrange(case_number):
        for j in xrange(case_number):
            if LA.norm(Y[i][j]) > 1e-5 and i!=j:
                X_p[i][j] = np.reciprocal(np.imag(np.reciprocal(Y[i][j])))
    for i in xrange(case_number):
        X_p[i][i] = g_b_round[i]
        for j in xrange(case_number):
            if i != j:
                X_p[i][i] -= X_p[i][j]

    for i in xrange(0, n_P):
        for j in xrange(0, n_P):
            B_p[i][j] = X_p[index_P[i]][index_P[j]]
    # ---------------------------------------------------
    for i in xrange(0, n_PQ):
        for j in xrange(0, n_PQ):
            B_pp[i][j] = B[index_PQ[i]][index_PQ[j]]

    return - B_p, B_pp

# Stott ================================================================================================================
# XX--------------------------------------------------------------------------------------------------------------------