Help with simplifying a code + np.savetxt()

[Laplace12]

Hey! I need help simplifying a code. I have a code which plots given data, so you'd only have to create a new data file and the basic Function code itself is the same for each. However, there is a lot of comment-uncomment stuff there that could probably be put in a more sensible manner. Here is the code for the plotting function:

def plot_data(energy, offset, area, width, center, figname):
    fig = plt.figure()
    #plt.errorbar(energy, offset[:,0], yerr=offset[:,1])
    plt.errorbar(energy, area[:,0], yerr=area[:,1])
    #plt.errorbar(energy, width[:,0], yerr=width[:,1])
    #plt.errorbar(energy, center[:,0], yerr=center[:,1])
    plt.ylabel('x')
    plt.xlabel('y')
    plt.legend()
    plt.savefig(figname, Transparent = True)

and here is a part of one data file:

energy=np.arange(1, n_pts+1)
area = np.zeros((n_pts, n_gaussian*4))
offset = np.zeros((n_pts, n_gaussian))
width = np.zeros((n_pts, n_gaussian*4))
center = np.zeros((n_pts, n_gaussian*4))
file = 'Name'
figname = '(the location of the fig in my computer)'
fn.plot_data(energy, offset, area, width, center, figname)

So what I'm wondering is if it is possible to simplify the plot_data function to plot any data that is given as input, so there would be no need to comment-uncomment every time I wish to plot different things? I know that varargin-argument would enable the function to accept any number of input arguments, so could that be of use here?

Another question I have is about the np.savetxt() command. I have two sets of data, transmitted beam intensity and initial beam intensity, I0 and I1, which I have defined like this,

name='I1'
scan_number=np.array([23, 24, 25, 26, 27, 28, 29, 30, 
                      31, 32, 33, 34, 35, 36, 37, 38, 39, 40])
n_pts = 701
n_chnl = 4096
mca_start = 3116

name='I0'
scan_number=np.array([45, 46, 47])
n_pts = 701
n_chnl = 4096
mca_start = 17138]

The point would be to get a file that I could use in data analysis, however even though the code works perfectly, I can't get it to save both I0 and I1 area in one file. I tried defining them separately (I0_area, I0_width etc.) and while that works, it's extremely tedious if there's more than two sets of data, so is there any simpler way to do this? Area, width, offset etc. are now defined once, but the file this creates has two identical area columns, when I wish to have one for I0 values and one for I1, so what I'm asking is if it's possible to have these values saved without defining them separately for each?

The part responsible for saving in my code looks like this (again, has to be simplified somehow)

def savetxt(area, width, offset, center, data, name, energy, file):
    np.savetxt(name + '.area', area)
    np.savetxt(name + '.width', width)
    np.savetxt(name + '.offset', offset)
    np.savetxt(name + '.center', center)
    np.savetxt(name + '.data', data)
    np.savetxt(file,
           np.transpose(([energy, area[:,0], area[:,0]])),
           header='Energy I0 I1 {}'.format(name))

Any help is greatly appreciated! I am very new to coding and all the comment-uncomment stuff makes sense to me as it is very simple, but doesn't look very nice on the code. Sp please note if you have any recommendations on where I could start with - which command etc.!