rasmus/nafuma
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Make sure filenames is a list before reading

Browse source
This commit is contained in:
rasmusvt 2022-06-29 16:13:19 +02:00
parent c0af1dc84c
commit 254becff69
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
nafuma/xanes/io.py
View file

@ -183,12 +183,13 @@ def read_data(data: dict, options={}) -> pd.DataFrame:
columns = ['ZapEnergy'] columns = ['ZapEnergy']
if not isinstance(data['path'], list):
data['path'] = [data['path']]
# Initialise DataFrame with only ZapEnergy-column # Initialise DataFrame with only ZapEnergy-column
xanes_data = pd.read_csv(data['path'][0])[['ZapEnergy']] xanes_data = pd.read_csv(data['path'][0])[['ZapEnergy']]
xanes_data['ZapEnergy'] += options['adjust'] xanes_data['ZapEnergy'] += options['adjust']
if not isinstance(data['path'], list):
data['path'] = [data['path']]
for filename in data['path']: for filename in data['path']:
columns.append(filename) columns.append(filename)