du-project/plotcurrent.py at master · Quantanalyst/du-project · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
## make plotting current automatic

## import libraries
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import os

class PlotCurrent():

    def __init__(self,device,number):
        self.device = device
        self.number = number

    def plotcurrent(self):

        ## check the device
        if (self.device == 'PCL' or self.device == 'PCT' or self.device == 'VISTA' or self.device == 'CB'):
            # print(os.getcwd())
            os.chdir("..")
            os.chdir("..")
            path = os.getcwd()
            # print(path)
            # read the info from HR info excel file
            infoloc = path + "\\Data from Dr\\BCM HIL PI Point List Draft 1 - 20161108.xlsx"
            measurement = pd.read_excel(infoloc, sheet_name='Measurement')

            ### read AI1 data set
            Fileloc = path + '\\tests\\3.1 AI-1.csv'
            AI1 = pd.read_csv(Fileloc)

            devicelist = measurement['Measurement Name'].str.split('_', 2).tolist()  ## list of device names and numbers
            hrlist = measurement['HR Number']  ## list of HR numbers

            hr_list = []
            for val in np.arange(0, len(devicelist)):
                if str(devicelist[val]) == 'nan':
                    pass
                elif devicelist[val][0] == str(self.device) and devicelist[val][1] == str(
                        self.number):  # if it finds a match, it will append the hr_list
                    hr_list.append(hrlist[val])

            ## creating a dataset of desired values to be plotted
            IrmsA = AI1[str(int(hr_list[0]))][1:]
            IrmsB = AI1[str(int(hr_list[1]))][1:]
            IrmsC = AI1[str(int(hr_list[2]))][1:]
            time = pd.DataFrame(np.arange(1, len(IrmsA) + 1))

            ############# plotting voltage  #############
            fig, axes = plt.subplots(nrows=1, ncols=3)  ## create a subplot
            # st = fig.suptitle("F_11 Breaker Voltage", fontsize="x-large")

            ## plot Phase A rms value
            axes[0].plot(time, IrmsA.apply(lambda x: int(x) / 100))
            axes[0].set_xlim([0, 300])
            #axes[0].set_ylim([0.7, 1.3])
            axes[0].set_xlabel('time(sec)')
            axes[0].set_ylabel('Phase A rms value')

            ## plot Phase B rms value
            axes[1].plot(time, IrmsB.apply(lambda x: int(x) / 100))
            axes[1].set_xlim([0, 300])
            #axes[1].set_ylim([0.7, 1.3])
            axes[1].set_xlabel('time(sec)')
            axes[1].set_ylabel('Phase B rms value')

            ## plot Phase C rms value
            axes[2].plot(time, IrmsC.apply(lambda x: int(x) / 100))
            axes[2].set_xlim([0, 300])
            #axes[2].set_ylim([0.7, 1.3])
            axes[2].set_xlabel('time(sec)')
            axes[2].set_ylabel('Phase C rms value')

            plt.tight_layout()
            plt.show()
            # fig.savefig("test.png")
        else:
            raise print('There is not any current value for this device')