From e6a4e2c81f5c72544d3894b925fa4faf8eac520f Mon Sep 17 00:00:00 2001 From: rasmusvt Date: Tue, 2 Aug 2022 13:48:29 +0200 Subject: [PATCH] Allow deletion of certain datapoints from the datasets --- nafuma/electrochemistry/io.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/nafuma/electrochemistry/io.py b/nafuma/electrochemistry/io.py index 328cf24..d3d6ad9 100644 --- a/nafuma/electrochemistry/io.py +++ b/nafuma/electrochemistry/io.py @@ -266,7 +266,9 @@ def process_neware_data(df, options={}): 'molecular_weight': None, 'reverse_discharge': False, 'splice_cycles': None, - 'increment_cycles_from': None} # index + 'increment_cycles_from': None,# index + 'delete_datapoints': None, # list of indices + } aux.update_options(options=options, required_options=required_options, default_options=default_options) @@ -282,9 +284,15 @@ def process_neware_data(df, options={}): df = unit_conversion(df=df, options=options) # converts all units from the old units to the desired units + print(df.iloc[1:10]) + if options['increment_cycles_from']: df['cycle'].iloc[options['increment_cycles_from']:] += 1 + if options['delete_datapoints']: + for datapoint in options['delete_datapoints']: + df.drop(index=datapoint, inplace=True) + if options['splice_cycles']: df = splice_cycles(df=df, options=options)