Skip to content

Commit

Permalink
remove dependencies on ectoolkit
Browse files Browse the repository at this point in the history
  • Loading branch information
robinzyb committed Oct 23, 2023
1 parent 3b6449e commit 16dbc7e
Showing 1 changed file with 10 additions and 1 deletion.
11 changes: 10 additions & 1 deletion cp2kdata/plots/fep_plot.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,14 +2,23 @@
import matplotlib.pyplot as plt
import glob
import os
from toolkit.utils.utils import get_cum_mean
from cp2kdata.block_parser.fep import parse_vertical_gap
from scipy import integrate
from cycler import cycler
from matplotlib.ticker import (MultipleLocator, AutoMinorLocator)

au2eV = 2.72113838565563E+01

def get_cum_mean(array):
tot = 0.0
cum_mean_array = []
for idx, i in enumerate(array):
tot += i
cum_mean_array.append(tot/(idx+1))
cum_mean_array = np.array(cum_mean_array)
return cum_mean_array


def plot_ti(fig_name):
eta_sub_dir_list = glob.glob("[0-1].*")
eta_sub_dir_list.sort()
Expand Down

0 comments on commit 16dbc7e

Please sign in to comment.