zebra/pyzebra
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Convert Counts to float ndarray

Browse Source
This commit is contained in:
usov_i 2021-02-09 11:32:55 +01:00
parent bf3b44405d
commit a6f97f59e8
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pyzebra/ccl_io.py
View File

@ -152,8 +152,8 @@ def parse_1D(fileobj, data_type):
# subsequent lines with counts # subsequent lines with counts
counts = [] counts = []
while len(counts) < s["n_points"]: while len(counts) < s["n_points"]:
counts.extend(map(int, next(fileobj).split())) counts.extend(map(float, next(fileobj).split()))
s["Counts"] = counts s["Counts"] = np.array(counts)
# add metadata to each scan # add metadata to each scan
s["meta"] = metadata s["meta"] = metadata
@ -175,6 +175,9 @@ def parse_1D(fileobj, data_type):
for name, val in zip(col_names, line.split()): for name, val in zip(col_names, line.split()):
s[name].append(float(val)) s[name].append(float(val))
for name in col_names:
s[name] = np.array(s[name])
try: try:
s["h"], s["k"], s["l"] = map(float, metadata["title"].split()[-3:]) s["h"], s["k"], s["l"] = map(float, metadata["title"].split()[-3:])
except (ValueError, IndexError): except (ValueError, IndexError):