diff --git a/examples/example00.py b/examples/example00.py index 8f8dfd0..e98360e 100644 --- a/examples/example00.py +++ b/examples/example00.py @@ -45,7 +45,8 @@ for d in densities: # compute the corresponding averaged radius (in Bohr units) rs = ( (1/d) / ((4/3) * np.pi) ) ** (1/3) / a0 # Some output printing - print("Computing {} for RS={}".format(plot_type, rs)) + stem = 'plas_disp' + print("Computing {} for RS={}".format(stem, rs)) # Here is the actual call to calculate data = compute(N_Q=100, RS=rs, T=T) # 'data' is a dict, each key is the name of an output file (without extansion). @@ -53,7 +54,7 @@ for d in densities: # the number of points chosen and on the kind of output file. # Reading data - X, Y = data['plas_disp'].T + X, Y = data[stem].T # Plotting plt.plot(X, Y, label=r'density={:.2e} $e/cm^3$'.format(d))