Mbyja-Wind-Project/class_datos.py at main · SebasMP14/Mbyja-Wind-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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
import pandas as pd
import numpy as np
from datetime import datetime
from mpl_toolkits.axes_grid1 import host_subplot
import mpl_toolkits.axisartist as AA
import matplotlib.pyplot as plt

class datos():
  def __init__(self, year, month, day):
    self.year = year
    self.month = month
    self.day = day
    self.path = self.create_path()
    self.df = pd.DataFrame()
    self.create_dataset()

  def get_year(self):
    return self.year

  def set_year(self, new_year):
    self.year = new_year
    return

  def get_month(self):
    return self.month

  def set_month(self, new_month):
    self.month = new_month
    return

  def get_day(self):
    return self.day

  def set_day(self, new_day):
    self.day = new_day
    return

  def create_path(self):
    path = "https://sohoftp.nascom.nasa.gov/sdb/goes/ace/monthly/{year}{month}_ace_swepam_1h.txt".format(year = self.year, month = self.month)
    return path

  def get_dataset(self):
    return self.df

  def set_date(self, year, month, day):
    self.set_year(year)
    self.set_month(month)
    self.set_day(day)
    self.set_dataset()

  def set_dataset(self):
    self.path = self.create_path()
    self.create_dataset()

  def create_dataset(self):
    self.df = pd.read_fwf(self.path, skiprows = 18, header = None)
    cols = ["Year","Month","Day","Time","Modified Julian Day","Seconds of the Day",
            "S","Proton_Density","Bulk_Speed","Ion_Temperature"]
    self.df.columns = cols
    self.df = self.df.replace(to_replace=[-9999.9,-1.00e+05],value=[np.NaN, np.NaN]).ffill().bfill()
    self.df["Time"] = pd.to_datetime(self.df["Seconds of the Day"], unit='s').dt.strftime('%H:%M')
    self.df["Date"] = pd.to_datetime(dict(year = self.df["Year"],
                                    month = self.df["Month"],
                                    day = self.df["Day"]))#,
                                    #hours = self.df["Time"].apply(lambda x: x.split(":")[0])))
    index = pd.MultiIndex.from_frame(self.df[["Date","Time"]])
    self.df = self.df.set_index(index).drop(columns = ["Year", "Month", "Day","Date","Modified Julian Day","Seconds of the Day","Time"])
    #self.df = self.norm_dataset(self.df)
    return self.df

  def norm_dataset(self, df):
    return (df - df.min()) / ( df.max() - df.min())

  def get_values(self, opt):
    if opt == "density":
      request = "Proton_Density"
    elif opt == "speed":
      request = "Bulk_Speed"
    elif opt == "temperature":
      request = "Ion_Temperature"

    resp = self.df.iloc[self.df.index.get_level_values('Date') == "{year}-{month}-{day}".format(year = self.year, month = self.month, day = self.day)]
    resp = resp[request]
    return resp.values

  def visualize_data(self, opt):
    if opt == "density":
      request = "Proton_Density"
      label = "Proton Density [p/cc]"
    elif opt == "speed":
      request = "Bulk_Speed"
      label = "Bulk Speed [km/s]"
    elif opt == "temperature":
      request = "Ion_Temperature"
      label = "Ion Temperature [K]"

    fig, ax = plt.subplots()

    x_axis = self.df.iloc[self.df.index.get_level_values('Date') == "{year}-{month}-{day}".format(year = self.year, month = self.month, day = self.day)]
    horas = x_axis.reset_index()["Time"].apply(lambda x: x.split(":")[0]).astype(int)

    ax.plot(horas, x_axis[request])
    ax.set_xlabel("{year}/{month}/{day}".format(year = self.year, month = self.month, day = self.day) + " - Hours of the day", fontdict = {'fontsize':14, 'fontweight':'bold', 'color':'tab:blue'})
    ax.set_ylabel(label)
    ax.grid(axis = 'y', color = 'gray', linestyle = 'dashed')
    return fig