Y Matrix Python

answered Aug 5 '2

vnedelea
1 ●2 ●3 ●3

updated Aug 5 '2

jconto
2756 ●10 ●30 ●41

I have created a code to build the Y matrix. The problem is that I've noticed that the funtion outputYmatrix does not calculate the Y in all nodes in my case... someone knows why?

import numpy as np
import pandas as pd
import sqlite3
import re
import psse35
import psspy
import pssarrays

psspy.psseinit()

_i = psspy.getdefaultint()

_f = psspy.getdefaultreal()

_s = psspy.getdefaultchar()

psspy.case(r'CASE.sav')

psspy.output_y_matrix(_i,_i,_i,0,r'outYmatrix.txt')

f = open(r'outYmatrix.txt')

listaY = []

for lines in f:

    if lines.startswith(' ZERO') or not lines.strip(): break

    else:

        listaY.append(re.split('[,\n]',lines))


conn = sqlite3.connect(':memory:')

cursor = conn.cursor()

cursor.execute('CREATE TABLE YMATRIX (IBUS INT, JBUS INT, Yreal real, Yimag real)')

conn.commit()

[cursor.execute(f'INSERT INTO YMATRIX VALUES ({item[0]},{item[1]},{item[2]},{item[3]})')for item in listaY]

df = pd.read_sql('SELECT * FROM YMATRIX ORDER BY IBUS, JBUS',conn).set_index('IBUS')

df['Y'] = df.Yreal + 1j*df.Yimag

df = df.astype({'Y':str})

df2 = df.reset_index().groupby([df.index, 'JBUS'])['Y'].aggregate('first').unstack().fillna('(0+0j)')

Ymatrix=df2.to_numpy(complex)

print(Ymatrix.shape)

print(Ymatrix)