pv-current/visualizer.py

# Created by: leo
# Created on: 2018.10.30

import matplotlib.pyplot as plt
import pandas as pd
from max_current import string_coef_maxcur

inpfiles = ['ycz2505.csv', 'ycz3208.csv', 'ycz3602.csv', 'ycz3608.csv',
            'ycz3611.csv', 'ycz6502.csv', 'ycz6604.csv']
string_ids = [12, 16, 16, 2, 4, 1, 2]


for file, str_id in zip(inpfiles, string_ids):
    raw = pd.read_csv('data/' + file,  usecols=[0,4,6],
                      index_col = 0).dropna()
    raw.index = pd.to_datetime(raw.index)
    months = raw.groupby(pd.Grouper(freq='M'))
    month_max_current = months['value'].max()

    daily = raw.groupby(pd.Grouper(freq='D'))
    normal_cur = daily.filter(lambda x: x['value'].max() > 2).groupby(pd.Grouper(freq='D'))
    low_cur = daily.filter(lambda x: x['value'].max() <= 2).groupby(pd.Grouper(freq='D'))

    coefs = low_cur.apply(string_coef_maxcur, str_id)
    plt.scatter(coefs.index, coefs, alpha=0.5, s=10, marker='x')
    plt.xticks(rotation=70)
    # print(file)
    # print(coefs)
    # plt.show()

plt.xlabel('Date')
plt.ylabel('String Coeffients')
plt.title('Coeffients of Strings')
plt.grid()
plt.legend(inpfiles)
plt.show()