# Basic libraries
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
from urllib import request

# function for downloading and storing data into variables.
def get_data(shot,identifier,*argv):
    try:
        data = pd.read_csv(f'http://golem.fjfi.cvut.cz/shots/{shot}/DASs/StandardDAS/{identifier}.csv',
                          names = ['t','data'])
    except:    
        data.t = data.t*1000
        data.data -= data.data[0:10].mean()
    return data

shot_no = 34251
Uloop = get_data(shot_no,'LoopVoltageCoil_raw')
fig,ax = plt.subplots(1)
Uloop[['t', 'data']].plot(x='t',ax=ax)
# ax.plot(Uloop_vac.t,Uloop_vac.data)
ax.set_xlabel('t [ms]')
ax.set_ylabel(r'$U_\mathrm{loop}$ [V]')
ax.grid(True)
plt.savefig('Uloop')