From ee4c5d3cd512e905fce230965e9e7836015fa30e Mon Sep 17 00:00:00 2001 From: Daniel Friesel Date: Wed, 15 Jun 2022 08:43:29 +0200 Subject: shift_param_in_aggregate: add '*=...' support --- lib/utils.py | 22 +++++++++++++++------- 1 file changed, 15 insertions(+), 7 deletions(-) (limited to 'lib') diff --git a/lib/utils.py b/lib/utils.py index 77f9954..12bd7c2 100644 --- a/lib/utils.py +++ b/lib/utils.py @@ -433,13 +433,21 @@ def shift_param_in_aggregate(aggregate, parameters, parameter_shift): :param parameter_filter: [[name, value], [name, value], ...] list of parameter values to keep, all others are removed. Values refer to normalizad parameter data. """ for param_name, param_shift_function in parameter_shift: - param_index = parameters.index(param_name) - for name in aggregate.keys(): - for param_list in aggregate[name]["param"]: - if param_list[param_index] is not None: - param_list[param_index] = param_shift_function( - param_list[param_index] - ) + if param_name == "*": + for name in aggregate.keys(): + for param_list in aggregate[name]["param"]: + for param_index in range(len(param_list)): + param_list[param_index] = param_shift_function( + param_list[param_index] + ) + else: + param_index = parameters.index(param_name) + for name in aggregate.keys(): + for param_list in aggregate[name]["param"]: + if param_list[param_index] is not None: + param_list[param_index] = param_shift_function( + param_list[param_index] + ) def filter_aggregate_by_param(aggregate, parameters, parameter_filter): -- cgit v1.2.3