From 254becff69cbf7bca4d4f5a0e93a01b10c8c29a1 Mon Sep 17 00:00:00 2001 From: rasmusvt Date: Wed, 29 Jun 2022 16:13:19 +0200 Subject: [PATCH] Make sure filenames is a list before reading --- nafuma/xanes/io.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/nafuma/xanes/io.py b/nafuma/xanes/io.py index 9676608..be7c9f3 100644 --- a/nafuma/xanes/io.py +++ b/nafuma/xanes/io.py @@ -183,12 +183,13 @@ def read_data(data: dict, options={}) -> pd.DataFrame: columns = ['ZapEnergy'] + if not isinstance(data['path'], list): + data['path'] = [data['path']] + # Initialise DataFrame with only ZapEnergy-column xanes_data = pd.read_csv(data['path'][0])[['ZapEnergy']] xanes_data['ZapEnergy'] += options['adjust'] - if not isinstance(data['path'], list): - data['path'] = [data['path']] for filename in data['path']: columns.append(filename)